Showing
2 changed files
with
6 additions
and
12 deletions
... | @@ -246,12 +246,9 @@ if __name__ == "__main__": | ... | @@ -246,12 +246,9 @@ if __name__ == "__main__": |
246 | oFile.write('Classifier: {}\n'.format(args.classifier)) | 246 | oFile.write('Classifier: {}\n'.format(args.classifier)) |
247 | oFile.write('Kernel: {}\n'.format(args.kernel)) | 247 | oFile.write('Kernel: {}\n'.format(args.kernel)) |
248 | oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) | 248 | oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) |
249 | - #oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted'))) | 249 | + oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted'))) |
250 | - #oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted'))) | 250 | + oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted'))) |
251 | - #oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted'))) | 251 | + oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted'))) |
252 | - oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred))) | ||
253 | - oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred))) | ||
254 | - oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred))) | ||
255 | oFile.write('Confusion matrix: \n') | 252 | oFile.write('Confusion matrix: \n') |
256 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') | 253 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') |
257 | oFile.write('Classification report: \n') | 254 | oFile.write('Classification report: \n') | ... | ... |
... | @@ -292,12 +292,9 @@ if __name__ == "__main__": | ... | @@ -292,12 +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=None))) | 295 | + oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted'))) |
296 | - oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average=None))) | 296 | + oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted'))) |
297 | - oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average=None))) | 297 | + oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted'))) |
298 | - #oFile.write('Precision: {}\n'.format(precision_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))) | ||
301 | oFile.write('Confusion matrix: \n') | 298 | oFile.write('Confusion matrix: \n') |
302 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') | 299 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') |
303 | oFile.write('Classification report: \n') | 300 | oFile.write('Classification report: \n') | ... | ... |
-
Please register or login to post a comment