第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)
NameError Traceback most recent call lastCell In18 line 5 3 y = data类别 4 x_train x_test y_train y_test = train_test_splitx y test_size=03 random_state=70----

原文地址: https://www.cveoy.top/t/topic/hrIG 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录