labeling update
This commit is contained in:
parent
d4b0de35d4
commit
97c2bde73d
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
|
@ -170,10 +170,10 @@ class MNBInteractive:
|
||||||
predictions_test = classifier.predict(testing_data_r)
|
predictions_test = classifier.predict(testing_data_r)
|
||||||
|
|
||||||
#print and store metrics
|
#print and store metrics
|
||||||
rec = recall_score(y[test], predictions_test, average=None)
|
rec = recall_score(y[test], predictions_test, average='macro')
|
||||||
print('rec: ' + str(rec))
|
print('rec: ' + str(rec))
|
||||||
recall_scores.append(rec)
|
recall_scores.append(rec)
|
||||||
prec = precision_score(y[test], predictions_test, average=None)
|
prec = precision_score(y[test], predictions_test, average='macro')
|
||||||
print('prec: ' + str(prec))
|
print('prec: ' + str(prec))
|
||||||
print('#')
|
print('#')
|
||||||
precision_scores.append(prec)
|
precision_scores.append(prec)
|
||||||
|
|
Loading…
Reference in New Issue