• Andrey Stebenkov's avatar
    Boosting method implementation (Catboost) (#1155) · 58e340cd
    Andrey Stebenkov authored
    Taking out Catboost model from SkLearnEvaluationStrategy into separate Boosting class strategy.
    
    New features for CatBoost:
    
    Converting data into Pool;
    Fitting with eval_set, early_stopping_rounds and best_model;
    Defining categoricat features for model;
    Ploting features importances;
    Saving/Loading model (required for implementation in Bagging method implementation to FEDOT #1005);
    Updated search space.
    Updates in data, preprocessing and data_split:
    
    Storing not encoded categorical features (into OHE or LE) and using it where necessary;
    Storing positions for different types of data (numeric, category and encoded);
    Storing features names of dataset.
    58e340cd
.readthedocs.yml 512 Bytes