Michael-Geis commited on
Commit
8573c13
1 Parent(s): c8a32d4

bugfix -load w/o specifying dataset

Browse files
Files changed (1) hide show
  1. embedding.py +6 -1
embedding.py CHANGED
@@ -14,7 +14,12 @@ class Embedder(BaseEstimator, TransformerMixin):
14
  return self
15
 
16
  def transform(
17
- self, X, y=None, model_name=None, load_from_file=False, path_to_embeddings=None
 
 
 
 
 
18
  ):
19
  """Either generates embeddings from an clean ArXivData instance or loads embeddings from file.
20
 
 
14
  return self
15
 
16
  def transform(
17
+ self,
18
+ X=None,
19
+ y=None,
20
+ model_name=None,
21
+ load_from_file=False,
22
+ path_to_embeddings=None,
23
  ):
24
  """Either generates embeddings from an clean ArXivData instance or loads embeddings from file.
25