【问题标题】:Error message when I try to do train test split on credit card default data当我尝试对信用卡默认数据进行训练测试拆分时出现错误消息
【发布时间】:2020-04-16 23:14:26
【问题描述】:

我尝试对来自https://archive.ics.uci.edu/ml/datasets/default+of+credit+card+clients# 的信用卡默认数据进行训练测试拆分

这是我的代码:

import sklearn
import pandas as pd

data = pd.read_excel("default of credit card clients.xls", sep=";")

x = data.drop(columns=['ID', 'default payment next month'], axis=1)

y = data['default payment next month']

x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(test_size=0.1)

当我尝试运行它时,我收到以下消息:

File "C:\Users\Kizo\Anaconda3\envs\tensorflow\lib\site-packages\sklearn\model_selection\_split.py", 
line 2086, in train_test_split
     raise ValueError("At least one array required as input")
ValueError: At least one array required as input

在我看来,x & y 没有传递给训练测试拆分函数,因为当我打印变量时,它们看起来像数组。请帮忙!

【问题讨论】:

    标签: python arrays train-test-split


    【解决方案1】:

    正如您所说的那样,xy 没有在您提供的代码行中传递。您只传递了参数 test_size 的值,但缺少 xy 值。你应该试试:

    x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(x,y,test_size=0.1)
    

    您还可以更改导入以简化代码:

    from sklearn.model_selection import train_test_split
    

    供以后使用:

    x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.1)
    

    【讨论】:

      猜你喜欢
      • 2020-06-17
      • 2018-08-11
      • 1970-01-01
      • 2020-02-03
      • 2018-04-30
      • 2016-02-27
      • 1970-01-01
      • 2019-08-14
      • 2022-01-05
      相关资源
      最近更新 更多