“Namhyun-Kim” commited on
Commit
87b6361
·
1 Parent(s): dc6f94e

Fix t-SNE call for older sklearn (fallback without n_iter)

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -240,14 +240,18 @@ def plot_tsne(tech_filter, snr_filter, mod_filter, mob_filter, representation, c
240
 
241
  eff_perplexity = min(perplexity, len(sampled_df) - 1)
242
  eff_perplexity = max(5, eff_perplexity)
243
- tsne = TSNE(
244
  n_components=2,
245
  perplexity=eff_perplexity,
246
- n_iter=n_iter,
247
  random_state=42,
248
  init="pca",
249
  learning_rate="auto",
250
  )
 
 
 
 
 
251
  try:
252
  projections = tsne.fit_transform(features)
253
  except Exception as exc:
 
240
 
241
  eff_perplexity = min(perplexity, len(sampled_df) - 1)
242
  eff_perplexity = max(5, eff_perplexity)
243
+ tsne_kwargs = dict(
244
  n_components=2,
245
  perplexity=eff_perplexity,
 
246
  random_state=42,
247
  init="pca",
248
  learning_rate="auto",
249
  )
250
+ try:
251
+ tsne = TSNE(**tsne_kwargs, n_iter=n_iter)
252
+ except TypeError:
253
+ # Older sklearn variants may not accept n_iter; fall back to defaults.
254
+ tsne = TSNE(**tsne_kwargs)
255
  try:
256
  projections = tsne.fit_transform(features)
257
  except Exception as exc: