| 
 | 
 
 
发表于 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, 以全面检测模型性能
 
 
  复制代码 |   
 
 
 
 |