Skip to content

Commit 6e4c245

Browse files
committed
* Scale parameters before clustering second time
* Pick clusters based on mean correlation instead of number of members.
1 parent 01eaeff commit 6e4c245

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tuneta/optimize.py

+4
Original file line numberDiff line numberDiff line change
@@ -345,6 +345,10 @@ def fit(self, X, y, idx=0, max_clusters=10, verbose=False, early_stop=None):
345345
num_clusters = None
346346
if num_clusters is None:
347347
num_clusters = int(len(params) * 0.2)
348+
349+
# Scale parameters for clustering
350+
params = mms.fit_transform(params)
351+
348352
if "index" in locals():
349353
kmeans = KPrototypes(n_clusters=num_clusters, random_state=42).fit(
350354
params, categorical=[index]

0 commit comments

Comments
 (0)