src.solutions.base_solution
Module Contents
Classes
Base class for any competition solution. |
- class src.solutions.base_solution.BaseSolution(config: Optional[dict] = None)
Base class for any competition solution.
- abstract fit(X: pandas.DataFrame, y: pandas.DataFrame, **kwargs)
- abstract predict(X: pandas.DataFrame) pandas.DataFrame
- abstract save(directory: Union[str, pathlib.Path]) None
Stores model to the directory.
The directory must be empty.
- abstract load(directory: Union[str, pathlib.Path]) None
Loads model from the directory.
Initializes the solution correctly even if the config dict wasn’t specified in the constructor.
The directory should not contain files other than those produced by the save method of the same class.