Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

...@@ -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
......