|
发表于 2021-5-21 22:39:32
|
显示全部楼层
你这代码都是 python2 的语法
试试这样
- from data import get_train_test_split
- from model import *
- from sklearn.metrics import accuracy_score,confusion_matrix,recall_score, f1_score
- from sklearn.model_selection import cross_val_score
- pipeline = get_naive_bayes_model()
- X_train, X_test, y_train, y_test = get_train_test_split()
- pipeline.fit(X_train, y_train)
- cv_scores = cross_val_score(pipeline, X_train, y_train, cv=5)
- print(cv_scores) # 这个地方报错
- predicted = pipeline.predict(X_test) # ['spam', 'ham']
- print(confusion_matrix(y_test, predicted))
- print('accuracy_score', accuracy_score(y_test, predicted))
- print('f1_score', f1_score(y_test, predicted))
- print('recall_score', recall_score(y_test, predicted)) #输出confusion matrix,并加入cross_validation, 以全面检测模型性能
复制代码 |
|