現状
機械学習のcross validationのセットの作成に取り組んでいます。
train.csvに訓練データがあり、folds.csvにfold番号を割り振っています。
folds.csv
fold 3 2 4 2 1 4 4 4 2 0 1 3 (以下略)
sklearnのtrain_test_splitを使えばすぐなのですが、自分で実装しているところです。
質問としては
folds = pd.read_csv("folds.csv") ids = folds[folds["fold"]==2].index
としてindex番号を取り出し、
train[ids]
とするとエラーが出ました。このような時どうやって取り出せば良いのでしょうか
よろしくお願いします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/01/24 02:44