Showing
1 changed file
with
2 additions
and
0 deletions
... | @@ -163,6 +163,8 @@ if __name__ == "__main__": | ... | @@ -163,6 +163,8 @@ if __name__ == "__main__": |
163 | oFile.write(str(confusion_matrix(trueEvaluationClasses, y_pred)) + '\n') | 163 | oFile.write(str(confusion_matrix(trueEvaluationClasses, y_pred)) + '\n') |
164 | oFile.write('Classification report: \n') | 164 | oFile.write('Classification report: \n') |
165 | oFile.write(classification_report(trueEvaluationClasses, y_pred) + '\n') | 165 | oFile.write(classification_report(trueEvaluationClasses, y_pred) + '\n') |
166 | + if options.classifier == "MLPClassifier": | ||
167 | + oFile.write("".format(classifier.coefs_)) | ||
166 | print(" Saving test report done!") | 168 | print(" Saving test report done!") |
167 | 169 | ||
168 | print("Training and test done in: %fs" % (time() - t0)) | 170 | print("Training and test done in: %fs" % (time() - t0)) | ... | ... |
-
Please register or login to post a comment