Python,機械学習,scikit-learnについての質問です。

Question

Python,機械学習,scikit-learnについての質問です。

自作のデータセットを用いて二値分類を行おうとしています。

Python3入門ノートという本のサンプルコードを真似して、
自作のexcelデータをShuffleSplitにかけようとした所でエラーが生じてしまいます。
どう検索しても同じようなエラーの方を見つけられずに困っています。
間違っている部分をどう書けばいいか教えてくださると幸いです。

エラーメッセージは以下のようなものです。
"None of [Int64Index([ 85, 6,154・・・・1795],\n dtype='int64', length=1265)] are in the [columns]"

Excelのデータは以下のようなデータセットを作成しました。

通電回数 通電時間 判定
660 23934 １
97 3307 ０
1444 43800 １
：
：
：
3695 230700 １

以下に、自分の書いたコードを記述させていただきます。

import pandas as pd
import numpy as np
from sklearn.model_selection import ShuffleSplit

df=pd.read_excel('python用サンプル.xlsx',sheetname='サンプル')
print(df.head())

X_col_names=['通電回数','通電時間']
y_col_name=['判定']

X=df[X_col_names]
y=df[y_col_name]

ss = ShuffleSplit(n_splits=1, train_size=0.6, test_size=0.4, random_state=0)
train_index, test_index = next(ss.split(X))

X_train, y_train = X[train_index], y[train_index]←ここでエラー
X_test, y_test = X[test_index], y[test_index]

お手数ですがご指南よろしくお願い致します。

Answer 1

DataFrame の一部の行を位置のインデックスで取得する場合、iloc を使用する必要があります。

X_train, y_train = X.iloc[train_index], y.iloc[train_index]
X_test, y_test = X.iloc[test_index], y.iloc[test_index] 

Comments

[Kanameecho]

回答ありがとうございます！

返信が遅くなってしまい申し訳ありません????

ちょうどPCを修理に出してるため、環境が整い次第実行してみます！
ほんとうにありがとうございました(^^)