src.solutions.many_bert_solution

Module Contents

Classes

ManyBertWithHandcraftedFeaturePredictor

Base class for any competition solution.

Functions

main()

Attributes

spellcheck

src.solutions.many_bert_solution.spellcheck

class src.solutions.many_bert_solution.ManyBertWithHandcraftedFeaturePredictor(model_names: list, catboost_iter: int, saving_dir: str)

Bases: src.solutions.base_solution.BaseSolution

Base class for any competition solution.

transform_data(X: pandas.Series) → pandas.DataFrame

fit(X: pandas.DataFrame, y: pandas.DataFrame, **kwargs) → None

predict(X: pandas.DataFrame) → pandas.DataFrame

save(directory: Union[str, pathlib.Path]) → None

Stores model to the directory.

The directory must be empty.

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.

src.solutions.many_bert_solution.main()