Ошибка при сохранении модели

Question

Я использую конвейер для разработки функций, а затем разделение проверки модели поезда для оптимизации гиперпараметров. Я хочу сохранить свою лучшую модель для последующего использования. Я использую следующий код:

tvs = TrainValidationSplit(estimator=pipeline,
                          estimatorParamMaps=paramGrid,
                          evaluator=BinaryClassificationEvaluator(),
                          trainRatio=0.8)



model = tvs.fit(train) # model is the model with combination of parameters that performed best
model.save('models/logisticregression_best_model')

Я получаю следующую ошибку: AttributeError: у объекта 'Pipeline' нет атрибута '_transfer_param_map_to_java'

Answer 1

Я смог решить эту проблему, изменив код:

model.save('models/logisticregression_best_model')

к этому коду:

model.bestModel.write().overwrite().save('models/logisticregression_best_model')