Restore/confusion matrix

beckstev / MachineLearningSeminar

MachineLearningSeminar SS19 TU Dortmund

MIT License

0 stars 0 forks source link

Restore/confusion matrix #25

Closed FeGeyer closed 5 years ago

FeGeyer commented 5 years ago

Usage: cd scripts and then python evaluate.py <model_path> <encoder_path>

Restored confusion matrix changes
implemented decoding of class names
implemented normalizing of colorbar

Still one class is predicted with absolute probability, but the class itself changes. Ein Trainingsdurchlauf auf mini_dataset Anderer Trainingsdurchlauf auf mini_dataset

I think this is because of bad training, but i'm not absolutely sure.

beckstev commented 5 years ago

Please check the confusion matrix of the latest model!

confusion_matrix.pdf

beckstev commented 5 years ago

I also modified the init if-condition in evaluate.py. For now on, we do not need the additional keyword init. The script checks automatically if prediction.txt does exist inside model_path.