make attributes private

etna-team · Nov 8, 2024 · 7b8a5f7 · 7b8a5f7
1 parent ce5c3ed
commit 7b8a5f7
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/etna/transforms/encoders/mean_segment_encoder.py b/etna/transforms/encoders/mean_segment_encoder.py
@@ -10,22 +10,21 @@
 class MeanSegmentEncoderTransform(IrreversibleTransform):
     """Makes expanding mean target encoding of the segment. Creates column 'segment_mean'."""
 
-    idx = pd.IndexSlice
-    segment_column = "segment_column"
+    _segment_column = "segment_column"
     out_column = "segment_mean"
 
     def __init__(self):
         super().__init__(required_features=["target"])
-        self.mean_encoder = MeanEncoderTransform(
-            in_column=self.segment_column, mode="per-segment", out_column=self.out_column, smoothing=0
+        self._mean_encoder = MeanEncoderTransform(
+            in_column=self._segment_column, mode="per-segment", out_column=self.out_column, smoothing=0
         )
 
     def _add_segment_column(self, df):
         segments = df.columns.get_level_values("segment").unique()
         flatten_segments = np.repeat(segments.values[np.newaxis, :], len(df), axis=0)
         segment_values = pd.DataFrame(
             data=flatten_segments,
-            columns=pd.MultiIndex.from_product([segments, [self.segment_column]]),
+            columns=pd.MultiIndex.from_product([segments, [self._segment_column]]),
             index=df.index,
         )
         df = pd.concat([df, segment_values], axis=1).sort_index(axis=1)
@@ -46,7 +45,7 @@ def _fit(self, df: pd.DataFrame) -> "MeanSegmentEncoderTransform":
             Fitted transform
         """
         df = self._add_segment_column(df)
-        self.mean_encoder._fit(df)
+        self._mean_encoder._fit(df)
         return self
 
     def _transform(self, df: pd.DataFrame) -> pd.DataFrame:
@@ -71,8 +70,8 @@ def _transform(self, df: pd.DataFrame) -> pd.DataFrame:
             If there are segments that weren't present during training.
         """
         df = self._add_segment_column(df)
-        df_transformed = self.mean_encoder._transform(df)
-        df_transformed = df_transformed.drop(columns=[self.segment_column], level="feature")
+        df_transformed = self._mean_encoder._transform(df)
+        df_transformed = df_transformed.drop(columns=[self._segment_column], level="feature")
         return df_transformed
 
     def get_regressors_info(self) -> List[str]: