Skip to content

plot loss evaluation metrics F1-score AUC and confusion matrix on the end of epoch

License

Notifications You must be signed in to change notification settings

prvccy/KerasPlotCallback

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

KerasPlotCallback

plot loss evaluation metrics F1-score AUC and confusion matrix on the end of epoch

usage:

plot loss and metrics in keras.model.compile() as default
    e.g.  keras.model.compile(loss = 'mse', metrics = 'mae')
          keras.model.fit(callbacks = PrvKerasCbk() )

add confusion matrix: (loss must be binary_crossentropy or categorical_crossentropy)
    keras.model.fit(callbacks = PrvKerasCbk(datagens=[PCDGConfusionMatrix()]) )
    
save best model(min loss) and training curve(loss and metrics) in h5 file:
    keras.model.fit(callbacks = PrvKerasCbk(controllers=[PKCSaveModelAndResult(path='/path/')]) )

About

plot loss evaluation metrics F1-score AUC and confusion matrix on the end of epoch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages