TabularPredictor.set_model_best¶
- TabularPredictor.set_model_best(model: str, save_trainer: bool = False)[source]¶
设置调用 predictor.predict(data) 时默认使用的模型。默认情况下，这是验证分数最佳的模型，但这并非总是如此。如果手动设置，当进行进一步训练时（例如通过 fit_extra, refit_full 或 distill），此设置可能会被内部覆盖。
- 参数：
model (str) – 要设置为最佳的模型名称。如果模型不存在或无法推断，则会引发 AssertionError。
save_trainer (bool, 默认为 False) – 如果为 True，则 self._trainer 会保存新的 model_best 值，以便在将来从磁盘加载预测器时反映此更改。