Set predictor for continued training. It is not recommended for user to call this function. Please use init_model argument in engine.train() or engine.cv() instead.
(self, predictor)
| 1306 | "set free_raw_data=False when construct Dataset to avoid this.") |
| 1307 | |
| 1308 | def _set_predictor(self, predictor): |
| 1309 | """Set predictor for continued training. |
| 1310 | |
| 1311 | It is not recommended for user to call this function. |
| 1312 | Please use init_model argument in engine.train() or engine.cv() instead. |
| 1313 | """ |
| 1314 | if predictor is self._predictor: |
| 1315 | return self |
| 1316 | if self.data is not None or (self.used_indices is not None |
| 1317 | and self.reference is not None |
| 1318 | and self.reference.data is not None): |
| 1319 | self._predictor = predictor |
| 1320 | return self._free_handle() |
| 1321 | else: |
| 1322 | raise LightGBMError("Cannot set predictor after freed raw data, " |
| 1323 | "set free_raw_data=False when construct Dataset to avoid this.") |
| 1324 | |
| 1325 | def set_reference(self, reference): |
| 1326 | """Set reference Dataset. |
no test coverage detected