Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

......@@ -254,6 +254,7 @@ if __name__ == "__main__":
oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n')
oFile.write('Classification report: \n')
oFile.write(classification_report(y_test, y_pred) + '\n')
oFile.write('Best parameters: \n')
for param in sorted(best_parameters.keys()):
oFile.write("\t%s: %r\n" % (param, best_parameters[param]))
print(" Done!")
......
......@@ -302,6 +302,7 @@ if __name__ == "__main__":
oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n')
oFile.write('Classification report: \n')
oFile.write(classification_report(y_test, y_pred) + '\n')
oFile.write('Best parameters: \n')
for param in sorted(best_parameters.keys()):
oFile.write("\t%s: %r\n" % (param, best_parameters[param]))
......