Skip to content

Commit

Permalink
Merge pull request #9 from kolmogorov-lab/hotfix
Browse files Browse the repository at this point in the history
'cat cols added to clustering'
  • Loading branch information
egorshishkovets committed Mar 27, 2024
2 parents c70b1fa + 18af7ac commit 179303b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion abacus/splitter/split_builder.py
Expand Up @@ -91,7 +91,7 @@ def _binnarize(self, df: pd.DataFrame) -> pd.DataFrame:
lambda group: self._clusterize(
df=group,
strata="strata",
columns=self.params.cols,
columns=self.params.cols + self.params.cat_cols,
min_cluster_size=self.params.min_cluster_size,
)
)
Expand Down

0 comments on commit 179303b

Please sign in to comment.