Skip to content

Commit

Permalink
scaling removed
Browse files Browse the repository at this point in the history
  • Loading branch information
marchunter authored May 30, 2018
1 parent 43cb357 commit 7c43e95
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions sklearn_krr_mlp.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,12 +76,12 @@ def split_data(x_data, y_data, ids_data, sample_size, is_mean, is_sparse, ids, i

def scale_data(is_mean, x_train, x_test):
# Scale
scaler = StandardScaler(with_mean=is_mean)
#scaler = StandardScaler(with_mean=is_mean)
# fit only on training data
scaler.fit(x_train)
x_train = scaler.transform(x_train)
#scaler.fit(x_train)
#x_train = scaler.transform(x_train)
# apply same transformation to test data
x_test = scaler.transform(x_test)
#x_test = scaler.transform(x_test)
return x_train, x_test

def load_split_scale_data(x_datafile, y_datafile, ids, testfiles,
Expand Down

0 comments on commit 7c43e95

Please sign in to comment.