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