diff --git a/tab_automl/automl/models.py b/tab_automl/automl/models.py index fc65bd2..9850627 100644 --- a/tab_automl/automl/models.py +++ b/tab_automl/automl/models.py @@ -49,6 +49,7 @@ "MeanShift": MeanShift, "OPTICS": OPTICS, "SpectralClustering": SpectralClustering, - "GaussianMixture": GaussianMixture + "GaussianMixture": GaussianMixture, + "HierarchicalDivisiveClustering": HierarchicalDivisiveClustering } }