Python 将错误获取为";TypeError:输入数据不能是列表;在XGBoost型号的Streamlight上部署时
我正在尝试使用XGBoost模型得到Streamlight的预测Python 将错误获取为";TypeError:输入数据不能是列表;在XGBoost型号的Streamlight上部署时,python,streamlit,Python,Streamlit,我正在尝试使用XGBoost模型得到Streamlight的预测 def load_dataset(): df = pd.read_csv('txn.csv') return df df = load_dataset() #create X and y, X will be feature set and y is the label - LTV X = df.drop(['LTVCluster','m1_Revenue'],axis=1) y = df['LTVCluste
def load_dataset():
df = pd.read_csv('txn.csv')
return df
df = load_dataset()
#create X and y, X will be feature set and y is the label - LTV
X = df.drop(['LTVCluster','m1_Revenue'],axis=1)
y = df['LTVCluster']
def train_model(X_train,y_train):
#XGBoost Multiclassification Model
xgb_classifier = XGBClassifier(max_depth=5, learning_rate=0.1,objective= 'multi:softprob',n_jobs=-1)
xgb_classifier.fit(X_train, y_train)
return xgb_classifier
def test_model(model,X_test,y_test):
y_pred = model.predict(X_test)
return y_pred
def predictions(CustomerID, Recency, RecencyCluster, Frequency, FrequencyCluster, Revenue, RevenueCluster, OverallScore):
predictions = model.predict([[CustomerID, Recency, RecencyCluster, Frequency, FrequencyCluster, Revenue, RevenueCluster, OverallScore]])
当我运行此代码时,得到的错误如下:
TypeError:输入数据不能是列表。
请帮助我哪里出了问题。这与Streamlight无关。