5

我正在尝试将 tpot 与我在熊猫数据框中的输入一起使用。我不断收到错误:

TypeError: 输入类型不支持 ufunc 'isnan',并且根据转换规则 ''safe'' 无法安全地将输入强制转换为任何支持的类型

我相信这个错误是由于 isnan 无法处理我的数据结构,但我不确定如何以不同的方式格式化它。我有分类和连续输入以及连续输出的组合。这是具有类似数据的代码示例:

train_x=[[1,2,3],['test1','test2','test3'],[56.2,4.5,3.4]]
train_y=[[3,6,7]]
from tpot import TPOTRegressor

tpot=TPOTRegressor()

我是否必须以某种方式转换我的分类数据?dataframe.values 和 dataframe.as_matrix 给我的对象也给我一个错误。

4

1 回答 1

5

没错 - 您需要转换分类值。TPOT 假设所有数据都采用 scikit-learn 兼容格式,这意味着所有数据都是数字的。我们最近才添加了对缺失值的支持,尽管大多数 scikit-learn 算法也不接受缺失值的数据。

我在下面修改了您的示例,以展示如何使用 pandas 将分类值转换为数值。

import pandas as pd
from tpot import TPOTRegressor

train_x = pd.DataFrame()
train_x['a'] = [1,2,3,4]
train_x['b'] = ['test1','test2','test3','test4']
train_x['c'] = [56.2,4.5,3.4,6.7]

# This line one-hot encodes the categorical variables
train_x = pd.get_dummies(train_x).values
# Print train_x out to understand what one-hot encoding entails
print(train_x)

train_y = [3,6,7,9]

my_tpot = TPOTRegressor(cv=2)
my_tpot.fit(train_x, train_y)
于 2018-04-13T18:35:16.973 回答