如何创建一个在已经拟合初始数据集后继续训练新样本的Estimator?

我正在尝试根据我在Spark源代码中找到的例子创建自己的Estimator,参见DeveloperApiExample.scala

但在这个例子中,每次我调用Estimator中的fit()方法时,它都会返回一个新的Model

我希望能够再次拟合以训练尚未训练的更多样本。

我考虑在Model类中创建一个新方法来实现这一点。但我不确定这是否合理。需要注意的是,我的模型不需要重新处理整个数据集来训练新样本,并且我们不希望改变模型结构。


回答:

Spark ml Estimator的基础类定义在这里。如你所见,类方法fit是一个使用输入数据训练模型的简单调用。

你应该参考LogisticRegression类,特别是trainOnRows函数,其中输入是一个RDD,并且可选地有一个初始系数矩阵(训练模型的输出)。这将允许你在一个不同的数据集上迭代地训练模型。

为了实现你的需求,请记住,你选择的算法必须能够支持迭代更新。例如,广义线性模型、神经网络、树集成等。

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中创建了一个多类分类项目。该项目可以对…

发表回复

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