Fix typo where label should be used. Original files work fine because labels
is a global variable
This commit is contained in:
parent
9390d63056
commit
c0691dfad9
@ -53,7 +53,7 @@ def preprocess_data(X, scaler=None):
|
||||
X = scaler.transform(X)
|
||||
return X, scaler
|
||||
|
||||
def preprocess_labels(y, encoder=None, categorical=True):
|
||||
def preprocess_labels(labels, encoder=None, categorical=True):
|
||||
if not encoder:
|
||||
encoder = LabelEncoder()
|
||||
encoder.fit(labels)
|
||||
|
Loading…
Reference in New Issue
Block a user