【发布时间】:2021-12-30 20:02:10
【问题描述】:
我已经在 anaconda 中安装了 sklearn 模块,但是 train_test_split 显示错误。 “名称'x_train'未定义”
from tensorflow import keras
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Dense, Flatten, Dropout
from tensorflow.keras.optimizers import Adam
from tensorflow.keras.callbacks import TensorBoard
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
但是在代码中它会抛出错误
train_df = pd.read_csv(r'fashion-mnist_train.csv')
test_df = pd.read_csv(r'fashion-mnist_test.csv')
train_data = np.array(train_df, dtype='float32')
test_data = np.array(test_df, dtype='float32')
x_train = train_data[:, 1:] / 255
y_train = train_data[:, 0]
x_test = test_data[:, 1:] / 255
y_test = test_data[:, 0]
x_train, x_validate, y_train, y_validate = train_test_split(
x_train, y_train, test_size=0.2, random_state=12345,
)
当我使用 train test split 运行单元时,会发生以下错误:
NameError Traceback(最近一次调用最后一次) 在 2 3 x_train,x_validate,y_train,y_validate = train_test_split( ----> 4 x_train, y_train, test_size=0.2, random_state=12345, 5) NameError: name 'x_train' 没有定义
请帮忙!这是我在 ML 中的第一个项目
【问题讨论】:
-
在 NameError 消息中,传递给 train_test_split 的参数与代码不完全匹配:在参数 random_state 之后传递了一个值 5。
标签: python pandas tensorflow keras scikit-learn