Hugging Face's logo

Spaces:
Adapting
/
TrendFlow
Runtime error

App Files Community
1
Adapting commited on
Commit
6085381
·
1 Parent(s): 73a7480

1.2.0

Browse files
Files changed (1) hide show
  1. lrt/utils/functions.py +1 -1
lrt/utils/functions.py CHANGED
@@ -57,7 +57,7 @@ def __create_model__(model_ckpt):
57
  return tmp[0].cpu().detach().numpy(), tmp[1].cpu().detach().numpy()
58
  return ret
59
  elif model_ckpt == 'pca':
60
- pca = template.dimension_reduction[model_ckpt](0.8)
61
  return pca
62
 
63
  elif model_ckpt =='kmeans-euclidean':
 
57
  return tmp[0].cpu().detach().numpy(), tmp[1].cpu().detach().numpy()
58
  return ret
59
  elif model_ckpt == 'pca':
60
+ pca = template.dimension_reduction[model_ckpt](0.95)
61
  return pca
62
 
63
  elif model_ckpt =='kmeans-euclidean':