tinkoff-ai · martins0n · Jul 5, 2022 · Jul 1, 2022 · Jul 1, 2022 · Jul 4, 2022
diff --git a/README.md b/README.md
@@ -69,7 +69,7 @@ train_ts, test_ts = ts.train_test_split(test_size=HORIZON)
 
 Define transformations and model:
 ```python
-from etna.models import CatBoostModelMultiSegment
+from etna.models import CatBoostMultiSegmentModel
 from etna.transforms import DateFlagsTransform
 from etna.transforms import DensityOutliersTransform
 from etna.transforms import FourierTransform
@@ -95,7 +95,7 @@ transforms = [
 ]
 
 # Prepare model
-model = CatBoostModelMultiSegment()
+model = CatBoostMultiSegmentModel()
 ```
 
 Fit `Pipeline` and make a prediction:

diff --git a/benchmarks/perfomance/configs/pipeline/daily_case1.yaml b/benchmarks/perfomance/configs/pipeline/daily_case1.yaml
@@ -1,7 +1,7 @@
 _target_: etna.pipeline.Pipeline
 horizon: 14
 model:
-  _target_: etna.models.CatBoostModelMultiSegment
+  _target_: etna.models.CatBoostMultiSegmentModel
 transforms:
 - _target_: etna.transforms.TimeSeriesImputerTransform
   in_column: target

diff --git a/benchmarks/perfomance/configs/pipeline/daily_case2.yaml b/benchmarks/perfomance/configs/pipeline/daily_case2.yaml
@@ -1,7 +1,7 @@
 _target_: etna.pipeline.Pipeline
 horizon: 14
 model:
-  _target_: etna.models.CatBoostModelPerSegment
+  _target_: etna.models.CatBoostPerSegmentModel
 transforms:
 - _target_: etna.transforms.TimeSeriesImputerTransform
   in_column: target

diff --git a/benchmarks/perfomance/configs/pipeline/daily_case3.yaml b/benchmarks/perfomance/configs/pipeline/daily_case3.yaml
@@ -1,7 +1,7 @@
 _target_: etna.pipeline.Pipeline
 horizon: 14
 model:
-  _target_: etna.models.CatBoostModelMultiSegment
+  _target_: etna.models.CatBoostMultiSegmentModel
 transforms:
 - _target_: etna.transforms.TimeSeriesImputerTransform
   in_column: target

diff --git a/benchmarks/perfomance/configs/pipeline/daily_case6.yaml b/benchmarks/perfomance/configs/pipeline/daily_case6.yaml
@@ -1,7 +1,7 @@
 _target_: etna.pipeline.Pipeline
 horizon: 14
 model:
-  _target_: etna.models.CatBoostModelMultiSegment
+  _target_: etna.models.CatBoostMultiSegmentModel
 transforms:
 - _target_: etna.transforms.TimeSeriesImputerTransform
   in_column: target

diff --git a/benchmarks/perfomance/configs/pipeline/default.yaml b/benchmarks/perfomance/configs/pipeline/default.yaml
@@ -1,7 +1,7 @@
 _target_: etna.pipeline.Pipeline
 horizon: 14
 model:
-  _target_: etna.models.CatBoostModelMultiSegment
+  _target_: etna.models.CatBoostMultiSegmentModel
 transforms:
   - _target_: etna.transforms.LinearTrendTransform
     in_column: target

diff --git a/docs/source/commands.rst b/docs/source/commands.rst
@@ -29,7 +29,7 @@ Example of pipeline's config:
     _target_: etna.pipeline.Pipeline
     horizon: 4
     model:
-      _target_: etna.models.CatBoostModelMultiSegment
+      _target_: etna.models.CatBoostMultiSegmentModel
     transforms:
       - _target_: etna.transforms.LinearTrendTransform
         in_column: target
@@ -103,7 +103,7 @@ Example of pipeline's config:
     _target_: etna.pipeline.Pipeline
     horizon: 4
     model:
-      _target_: etna.models.CatBoostModelMultiSegment
+      _target_: etna.models.CatBoostMultiSegmentModel
     transforms:
       - _target_: etna.transforms.LinearTrendTransform
         in_column: target

diff --git a/etna/models/__init__.py b/etna/models/__init__.py
@@ -6,6 +6,8 @@
 from etna.models.base import PerSegmentModel
 from etna.models.catboost import CatBoostModelMultiSegment
 from etna.models.catboost import CatBoostModelPerSegment
+from etna.models.catboost import CatBoostMultiSegmentModel
+from etna.models.catboost import CatBoostPerSegmentModel
 from etna.models.holt_winters import HoltModel
 from etna.models.holt_winters import HoltWintersModel
 from etna.models.holt_winters import SimpleExpSmoothingModel