交差検証についての質問

Question

前提・実現したいこと

層化交差検証を行っている際にわからないエラーが出てしまいました。どのようにすれば改善されますでしょうか？

発生している問題・エラーメッセージ

ValueError: Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

該当のソースコード

# 目的変数と説明変数の定義
x_train = train.iloc[:,[0,1,3,4,5,6,7,8,9]]
y_train = train['kpl']
x_test = test

# 層化法によるｋ分割による交差検証
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import StratifiedKFold
from sklearn.model_selection import cross_val_score

kf = StratifiedKFold(n_splits=3, shuffle=True, random_state=1)

lr = LinearRegression()

print(cross_val_score(lr, x_train, y_train, cv=kf))

試したこと

x_trainとy_trainの中身を再確認しました。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[y_waiwai]

確認してどういう結果だったんでしょうか

[hayataka2049]

エラーは省略せずすべて記載してください。また、できるだけ再現可能な情報をご提示ください（そのまま実行できるコードにする、必要なデータを掲載するなど）。

Answer 1

ValueError: Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

質問文のエラーメッセージが抜粋されているので以下憶測含みますが（Tracebackの一行目から全文掲載していただくのがベターなので次回以降の質問ではそうしてください）、

おそらくStratifiedKFoldの内部から送出された例外でしょう。連続値のtargetでは層別できないというエラーです。

さしあたってはKFoldに置き換えれば解決するのではないでしょうか。

sklearn.model_selection.KFold — scikit-learn 0.21.3 documentation

Comments

[rikubon_]

ありがとうございました。次からはそうさせていただきます