簡體   English   中英

cross_validation.cross_val_score 返回什么樣的分數?

[英]What kind of scores are returned by cross_validation.cross_val_score?

我在 sci-kit 學習文檔中找不到此信息。 但是根據我得到的數字,分數看起來不像是均方誤差。

cross_val_score調用您傳入的估算器的.score()方法,返回的內容因估算器而異。 您必須查看每個估算器的文檔以找出相應的.score()方法返回的內容。 您可以使用scoring參數覆蓋此默認行為。 在此處記錄

我首先使用 cross_val_predict 計算預測值,然后使用 y_test 使用預測值通過 metrics.accuracy_score 獲得分數,從而避免了這個問題。

# Function that runs the requested algorithm and returns the accuracy metrics
def fit_ml_algo(algo, X_train, y_train, cv):

# One Pass
model = algo.fit(X_train, y_train)
acc = round(model.score(X_train, y_train) * 100, 2)

# Cross Validation 
train_pred = model_selection.cross_val_predict(algo, 
                                              X_train, 
                                              y_train, 
                                              cv=cv, 
                                              n_jobs = -1)
# Cross-validation accuracy metric
acc_cv = round(metrics.accuracy_score(y_train, train_pred) * 100, 2)

return train_pred, acc, acc_cv

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM