TabularPredictor.set_model_best

TabularPredictor.set_model_best(model: str, save_trainer: bool = False)[source]

设置默认使用的模型,当调用predictor.predict(data)时。 默认情况下,这是具有最佳验证分数的模型,但情况并非总是如此。 如果手动设置,如果发生进一步的训练,例如通过fit_extra、refit_full或distill,这可能会在内部被覆盖。

Parameters:
  • model (str) – 要设置为最佳的模型名称。如果模型不存在或无法推断,则会引发AssertionError。

  • save_trainer (bool, default = False) – 如果为True,self._trainer将与新的model_best值一起保存,以便在将来从磁盘加载predictor时反映出来。