Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing structural domain dataset

......@@ -292,12 +292,12 @@ if __name__ == "__main__":
oFile.write('Classifier: {}\n'.format(args.classifier))
oFile.write('Kernel: {}\n'.format(args.kernel))
oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred)))
#oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted')))
#oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted')))
#oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted')))
oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred)))
oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred)))
oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred)))
oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted')))
oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted')))
oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted')))
#oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred)))
#oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred)))
#oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred)))
oFile.write('Confusion matrix: \n')
oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n')
oFile.write('Classification report: \n')
......