Spaces:
Running
Running
Update embedder_lib.py
Browse files- embedder_lib.py +0 -4
embedder_lib.py
CHANGED
@@ -22,10 +22,6 @@ class Embedder(BaseEstimator, TransformerMixin):
|
|
22 |
|
23 |
class FeaturesSelector(BaseEstimator, TransformerMixin):
|
24 |
def fit(self, X, y=None):
|
25 |
-
nc, nf = X.shape
|
26 |
-
nf = min(nf, nc//5)
|
27 |
-
self.features_selector = SelectKBest(k = nf)
|
28 |
-
self.features_selector.fit(X, y)
|
29 |
return self
|
30 |
|
31 |
def transform(self, X: pd.DataFrame):
|
|
|
22 |
|
23 |
class FeaturesSelector(BaseEstimator, TransformerMixin):
|
24 |
def fit(self, X, y=None):
|
|
|
|
|
|
|
|
|
25 |
return self
|
26 |
|
27 |
def transform(self, X: pd.DataFrame):
|