Skip to content

Commit 36b8ca1

Browse files
SvenBarrayBARRAY Sven
andauthored
requirements.txt update and attribute name change (#152)
* edit requirements.txt to match INSTALL_REQUIRES from setup.py * changed 'affinity' parameter to 'metric' to follow sklearn.cluster.AgglomerativeClustering parameter rename --------- Co-authored-by: BARRAY Sven <[email protected]>
1 parent 3d649d6 commit 36b8ca1

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

examples/clustering/plot_dtw_boss.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def plot_dendrogram(model, **kwargs):
7676
fig, axes = plt.subplots(1, 3, figsize=(16, 8))
7777

7878
X, y = make_cylinder_bell_funnel(n_samples=n_samples, random_state=42,
79-
shuffle=False)
79+
shuffle=False)
8080
for k_axis, metric in enumerate(["Euclidean", "DTW", "BOSS"]):
8181
if metric == "DTW":
8282
dist_mat = create_dist_matrix(X, dtw)
@@ -89,7 +89,7 @@ def plot_dendrogram(model, **kwargs):
8989

9090
model = AgglomerativeClustering(compute_full_tree=True,
9191
compute_distances=True,
92-
n_clusters=3, affinity="precomputed",
92+
n_clusters=3, metric="precomputed",
9393
linkage="complete")
9494
cluster = model.fit_predict(dist_mat)
9595
score = round(homogeneity_score(labels_true=y, labels_pred=cluster), 2)

requirements.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
numpy>=1.17.5
2-
scipy>=1.3.3
3-
scikit-learn>=0.22.1
4-
joblib>=0.14
5-
numba>=0.48.0
1+
numpy>=1.22.4
2+
scipy>=1.8.1
3+
scikit-learn>1.2.0
4+
joblib>=1.1.1
5+
numba>=0.55.2

0 commit comments

Comments
 (0)