【发布时间】:2021-02-08 19:56:05
【问题描述】:
我尝试创建一个机器学习模型来进行一些预测,但我一直遇到一个绊脚石。即,代码似乎忽略了我给它的插补指令,导致以下错误:
ValueError: Input contains NaN, infinity or a value too large for dtype('float64').
这是我的代码:
import pandas as pd
import numpy as np
from sklearn.ensemble import AdaBoostRegressor
from category_encoders import CatBoostEncoder
from sklearn.compose import make_column_transformer
from sklearn.preprocessing import OneHotEncoder, StandardScaler
from sklearn.pipeline import make_pipeline
from sklearn.impute import SimpleImputer
data = pd.read_csv("data.csv",index_col=("Unnamed: 0"))
y = data.Installs
x = data.drop("Installs",axis=1)
strat = ["mean","median","most_frequent","constant"]
num_imp = SimpleImputer(strategy=strat[0])
obj_imp = SimpleImputer(strategy=strat[2])
# Set up the scaler
sc = StandardScaler()
# Set up Encoders
cb = CatBoostEncoder()
oh = OneHotEncoder(sparse=True)
# Set up columns
obj = list(x.select_dtypes(include="object"))
num = list(x.select_dtypes(exclude="object"))
cb_col = [i for i in obj if len(x[i].unique())>30]
oh_col = [i for i in obj if len(x[i].unique())<10]
# First Pipeline
imp = make_pipeline((num_imp))
enc_cb = make_pipeline((obj_imp),(cb))
enc_oh = make_pipeline((obj_imp),(oh))
# Col Transformation
col = make_column_transformer((imp,num),
(sc,num),
(enc_oh,oh_col),
(enc_cb,cb_col))
model = AdaBoostRegressor(random_state=(0))
run = make_pipeline((col),(model))
run.fit(x,y)
这是代码中用于复制目的的数据的link。你能说出什么是错的吗?感谢您的宝贵时间。
【问题讨论】:
-
请提供错误回溯:它可以帮助您找到错误的来源。
标签: python pandas scikit-learn data-science valueerror