auto-sklearn配置及使用

最新推荐文章于 2024-06-05 21:45:00 发布

余余余2020

最新推荐文章于 2024-06-05 21:45:00 发布

阅读量1.8k

点赞数

文章标签：机器学习 sklearn 人工智能

本文链接：https://blog.csdn.net/weixin_52561314/article/details/124227819

版权

1.安装遇到的坑

1）swig安装，macos下用brew装一下即可。

2）内存调整，autosklearn的model fit对内存要求较高，把limit调到300000，否则报错。

2.使用及探索

1）数据加载及库引入，还是用iris数据集

"""
Created on Sat Apr 16 15:26:21 2022

@author: johnny
"""
import autosklearn
from autosklearn.classification import AutoSklearnClassifier
from sklearn.metrics import accuracy_score


#数据引入
from sklearn.datasets import load_iris
data = load_iris()
x = data.data
y = data.target

from sklearn.model_selection import train_test_split
train_x,test_x,train_y,test_y = train_test_split(x,y,test_size=0.3,random_state=0)

2）模型配置

# define search
model = AutoSklearnClassifier(memory_limit=300000,
                              time_left_for_this_task=5*60,
                              per_run_time_limit=5