Python NameError: 'train_x' is not defined - 解决方法

Python NameError: 'train_x' is not defined - 解决方法

错误代码:

NameError                                 Traceback (most recent call last)
Cell In[18], line 5
      3 y = data['类别']
      4 x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=70)
----> 5 num_train = len(train_x)
      6 num_test = len(test_x)

NameError: name 'train_x' is not defined

错误分析:

第5行代码中出现了错误，因为train_x并没有在代码中被定义。

解决方法:

应该将train_x和test_x改为x_train和x_test，即：

num_train = len(x_train)
num_test = len(x_test)

修改后的代码:

import pandas as pd
from sklearn.model_selection import train_test_split

data = pd.read_csv('data.csv')
x = data.drop(['类别'], axis=1)
y = data['类别']
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=70)
num_train = len(x_train)
num_test = len(x_test)

总结:

Python 中 NameError: 'train_x' is not defined 错误通常是由于变量名拼写错误或变量未定义导致的。检查代码，确保变量名拼写正确，并且变量已在代码中定义。