Question

我正在研究本教程https://medium.com/@sarahcy/read-this-how-winners-create-life-changing-habits-that-actually-work-atomic-habits-by-james-ac7a3c6df911，目前正在尝试运行模型评估部分

class_labels = list(set(labels))
meu.display_model_performance_metrics(true_labels=y_test, predicted_labels=predictions, classes=class_labels)

我收到此错误

Prediction Confusion Matrix:
------------------------------
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/blabla/XAI/model_evaluation_utils.py", line 87, in display_model_performance_metrics
    classes=classes)
  File "/blabla/XAI/model_evaluation_utils.py", line 62, in display_confusion_matrix
    labels=level_labels), 
TypeError: __new__() got an unexpected keyword argument 'labels'

我该如何解决这个问题？

BR

编辑：这是出现问题的代码：

# part1
import pandas as pd 
import numpy as np 
import model_evaluation_utils as meu
import matplotlib.pyplot as plt
from collections import Counter
import shap
import eli5

import warnings
warnings.filterwarnings('ignore')
plt.style.use('fivethirtyeight')

shap.initjs()

#part 2
data, labels = shap.datasets.adult(display=True)
labels = np.array([int(label) for label in labels])

print(data.shape , labels.shape)
data.head()

#part 3
Counter(labels)

#part 4
cat_cols = data.select_dtypes(['category']).columns
data[cat_cols] = data[cat_cols].apply(lambda x: x.cat.codes)
data.head()

#part 5
from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(data, labels, test_size=0.3, random_state=42)
print(X_train.shape, X_test.shape)
X_train.head(3)

data_disp, labels_disp = shap.datasets.adult(display=True)
X_train_disp, X_test_disp, y_train_disp, y_test_disp = train_test_split(data_disp, labels_disp, test_size=0.3, random_state=42)
print(X_train_disp.shape, X_test_disp.shape)
X_train_disp.head(3)

#part 6
import xgboost as xgb
xgc = xgb.XGBClassifier(n_estimators=500, max_depth=5, base_score=0.5,
                        objective='binary:logistic', random_state=42)
xgc.fit(X_train, y_train)

#part 7
predictions = xgc.predict(X_test)
predictions[:10]

#part 8
class_labels = list(set(labels))
meu.display_model_performance_metrics(true_labels=y_test, predicted_labels=predictions, classes=class_labels)

model_evaluation_utils混淆矩阵出错

0 个答案: