Как получить результат обучения Apache Spark ML после каждой итерации - PullRequest
0 голосов
/ 01 марта 2019

Есть ли способ получить оценку оценки, например, RMSE в случае регрессии и точного отзыва после каждой итерации?Например, если у меня есть это: LogisticRegression lr = new LogisticRegression () .setMaxIter (10) .setRegParam (0.3) .setElasticNetParam (0.8);

Можно ли получить 10 баллов, если установлена ​​максимальная интеграциядо 10?Любой указатель будет высоко оценен.Я использую Spark 2.4 в Java.Спасибо.

1 Ответ

0 голосов
/ 01 марта 2019

После того, как вы тренировались, у вас есть LogisticRegressionModel (предположим, это называется model).model.summary() дает вам LogisticRegressionTrainingSummary, а model.summary().objectiveHistory() - массив потерь на каждом шаге.Я не верю, что вы можете получить точность / отзыв после каждой итерации.

Обратите внимание, что не обязательно будет maxIter итераций - модель могла сойтись до этого.

...