search

FlorentF9 / DeepTemporalClustering

:chart_with_upwards_trend: Keras implementation of the Deep Temporal Clustering (DTC) model
MIT License
220 stars 58 forks source link

fix bug when y_train is None #9

Closed yarson closed 4 years ago

yarson commented 4 years ago

fields 'acc', 'pur', 'nmi', 'ari' are not defined when y_train is None, as function fit() shown:

...
if y_train is not None:
    fieldnames += ['acc', 'pur', 'nmi', 'ari']
...
FlorentF9 commented 4 years ago

You're right, thanks!