name 'train_test_split' is not definedのエラーが出てしまいます

Question

以下の様なエラーが出てしまいます。
これはどのような意味なのでしょうか。
ご教授ください。

Traceback (most recent call last)
<ipython-input-1-fa5627863ca1> in <module>
----> 1 X_train, X_test, Y_train, Y_test = train_test_split(df_xy[["SMA5/current", "SMA20/current","RSI","MACD","BBANDS+2σ","BBANDS-2σ"]],df_xy["前日比_classified"], train_size=0.8)

NameError: name 'train_test_split' is not defined

Answer 1

'train_test_split' という名前の関数が存在しない、といってますね

Comments

[maruSANKU]

ありがとうございます。

Answer 2

以下の例を見てください。

>>> print(x + 10)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'x' is not defined

なぜこういうエラーが出るかというと、変数xに値が入っていない、からです。

Pythonの場合は、変数に値を入れることを束縛(bind)といいます。
公式ドキュメント 4.2. 名前づけと束縛 (naming and binding)
には、以下のように書かれています。

以下の構造で、名前が束縛されます:

• 関数の仮引数 (formal parameter) 指定、
• import 文、
• クラスや関数の定義 (定義を行ったブロックで、クラスや関数名を束縛します)、
• 代入が行われるときの代入対象の識別子、
• for ループのヘッダ、
• with 文や except 節の as の後ろ。

上の例では、代入が行われるときの代入対象の識別子を使って、以下のように書くとエラーは出なくなります。

>>> x = 32
>>> print(x + 10)
42

ご質問の件も上記のいずれかの方法でtrain_test_splitという名前に何かを束縛することで解決できます。
そのために、maruSANKUさんが参考にしたコードを良く読み直して、該当するものを探してみることをお勧めします。

Comments

[maruSANKU]

ありがとうございます。確認してみます。

Answer 3

sklearn.model_selection.train_test_split

Comments

[maruSANKU]

ありがとうございます。見てみます。