如何修复sklearn fit_transform中的“元组索引超出范围”错误?

时间:2019-06-22 22:49:26

标签: scikit-learn tuples

我有一个一维元组,其中包含我从.csv文件读取的数据。然后,我使用下面的代码初始化One Hot Encoder类。但是,我不断收到错误消息

我尝试过将元组转换为2D元组,但是没有用。

@NgModule({
    providers: [...],
    declarations: [...],
    entryComponents: [DialogNamePromptComponent, ... ],
    ...
})
export class YourModule { }

我不断收到“ IndexError:元组索引超出范围”的错误

2 个答案:

答案 0 :(得分:0)

兄弟一切都很完美,但是您需要重塑数据集。

通过添加该行代码

  

output_data = temp.reshape(-1,1)

output_data = dataset.iloc[:,9].values

#Encoding the categorical output data (There is no categorical input data)
one_hot_encoder = OneHotEncoder(categorical_features = [0])

#Reshape dataset before apply fit_transform function
output_data = temp.reshape(-1, 1)
output_data = one_hot_encoder.fit_transform(output_data).toarray()

答案 1 :(得分:0)

使用get_dummies适用于一维

import numpy as np
y = pd.get_dummies(y)

记住将y转换回数组:

y = np.array(y)