Spaces:
Runtime error
Runtime error
Michael-Geis
commited on
Commit
•
8573c13
1
Parent(s):
c8a32d4
bugfix -load w/o specifying dataset
Browse files- embedding.py +6 -1
embedding.py
CHANGED
@@ -14,7 +14,12 @@ class Embedder(BaseEstimator, TransformerMixin):
|
|
14 |
return self
|
15 |
|
16 |
def transform(
|
17 |
-
self,
|
|
|
|
|
|
|
|
|
|
|
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 |
|