聚合国内IT技术精华文章,分享IT技术精华,帮助IT从业人士成长

The first trial for PyCaret

2020-06-26 10:18 浏览: 1850137 次 我要评论(0 条) 字号:

The code for using PyCaret is quite simple:

df = pd.read_csv(TRAIN_CSV_FILE)
setup(data=data, target="TARGET", session_id=1023)
compare_models(verbose = False)

But it reported error in the first run:

ValueError: array is too big; `arr.size * arr.dtype.itemsize` is larger than the maximum possible size.

This is because it has a custom_id column in my data that have string value like “11100000033445566”. Seems PyCaret recognize it as integer type.

After drop that column, I got the result of models comparison:

Looks the GradientBoostingClassifier works as well as LightGBM.



网友评论已有0条评论, 我也要评论

发表评论

*

* (保密)

Ctrl+Enter 快捷回复