Training, crossvalidation and testing structural domain dataset
Showing
1 changed file
with
3 additions
and
3 deletions
... | @@ -292,9 +292,9 @@ if __name__ == "__main__": | ... | @@ -292,9 +292,9 @@ if __name__ == "__main__": |
292 | oFile.write('Classifier: {}\n'.format(args.classifier)) | 292 | oFile.write('Classifier: {}\n'.format(args.classifier)) |
293 | oFile.write('Kernel: {}\n'.format(args.kernel)) | 293 | oFile.write('Kernel: {}\n'.format(args.kernel)) |
294 | oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) | 294 | oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) |
295 | - oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted'))) | 295 | + oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='micro'))) |
296 | - oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted'))) | 296 | + oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='micro'))) |
297 | - oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted'))) | 297 | + oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='micro'))) |
298 | #oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred))) | 298 | #oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred))) |
299 | #oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred))) | 299 | #oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred))) |
300 | #oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred))) | 300 | #oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred))) | ... | ... |
-
Please register or login to post a comment