Milad Alshomary
commited on
Commit
·
f310e45
1
Parent(s):
685033a
updates
Browse files
utils/clustering_utils.py
CHANGED
|
@@ -150,7 +150,8 @@ def clustering_author(background_corpus_df: pd.DataFrame,
|
|
| 150 |
|
| 151 |
if eps_values is None:
|
| 152 |
if metric == 'cosine':
|
| 153 |
-
eps_values = [0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8]
|
|
|
|
| 154 |
else: # 'euclidean' or other
|
| 155 |
if X.shape[0] > 1:
|
| 156 |
# For Euclidean, eps depends on the scale of the data.
|
|
|
|
| 150 |
|
| 151 |
if eps_values is None:
|
| 152 |
if metric == 'cosine':
|
| 153 |
+
#eps_values = [0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8]
|
| 154 |
+
eps_values = np.arange(0.01, 0.3, 0.01)
|
| 155 |
else: # 'euclidean' or other
|
| 156 |
if X.shape[0] > 1:
|
| 157 |
# For Euclidean, eps depends on the scale of the data.
|