Showing
2 changed files
with
2 additions
and
0 deletions
| ... | @@ -254,6 +254,7 @@ if __name__ == "__main__": | ... | @@ -254,6 +254,7 @@ if __name__ == "__main__": |
| 254 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') | 254 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') |
| 255 | oFile.write('Classification report: \n') | 255 | oFile.write('Classification report: \n') |
| 256 | oFile.write(classification_report(y_test, y_pred) + '\n') | 256 | oFile.write(classification_report(y_test, y_pred) + '\n') |
| 257 | + oFile.write('Best parameters: \n') | ||
| 257 | for param in sorted(best_parameters.keys()): | 258 | for param in sorted(best_parameters.keys()): |
| 258 | oFile.write("\t%s: %r\n" % (param, best_parameters[param])) | 259 | oFile.write("\t%s: %r\n" % (param, best_parameters[param])) |
| 259 | print(" Done!") | 260 | print(" Done!") | ... | ... |
| ... | @@ -302,6 +302,7 @@ if __name__ == "__main__": | ... | @@ -302,6 +302,7 @@ if __name__ == "__main__": |
| 302 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') | 302 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') |
| 303 | oFile.write('Classification report: \n') | 303 | oFile.write('Classification report: \n') |
| 304 | oFile.write(classification_report(y_test, y_pred) + '\n') | 304 | oFile.write(classification_report(y_test, y_pred) + '\n') |
| 305 | + oFile.write('Best parameters: \n') | ||
| 305 | for param in sorted(best_parameters.keys()): | 306 | for param in sorted(best_parameters.keys()): |
| 306 | oFile.write("\t%s: %r\n" % (param, best_parameters[param])) | 307 | oFile.write("\t%s: %r\n" % (param, best_parameters[param])) |
| 307 | 308 | ... | ... |
-
Please register or login to post a comment