在数据集上分块训练SGDRegressor

在进行机器学习任务时,我需要处理相当大的数据集。因此,我无法一次性将整个数据集加载到我的算法中。我正在寻找一种方法,可以将数据集分部分地训练我的算法,简单地输入新的数据块是行不通的,因为我的算法会重新拟合,而不会考虑之前的示例。是否有某种方法可以让我向算法输入新数据,同时“记住”之前看到的数据?

编辑:我使用的算法是来自scikit-learn的SGDRegressor。

代码如下:

train = pd.read_csv(os.path.join(dir,"Train.csv"),chunksize = 5000)labels = pd.read_csv(os.path.join(dir,"Labels.csv"),chunksize = 5000)algo = SGDRegressor(n_iter = 75)print("looping for chunks in train")for chunk in train:    algo.fit(train,labels)

回答:

您可以使用partial_fit来向SGDRegressor输入部分训练数据。

请查看示例代码

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注