Resolved a few minor issues found during testing.

This commit is contained in:
John Wittenauer 2015-08-18 21:05:45 -04:00
parent 7c966439fa
commit 4afb5b60d6
2 changed files with 6 additions and 6 deletions

@ -234,7 +234,7 @@ class KerasRegressor(BaseWrapper):
preds : array-like, shape = (n_samples)
Predictions.
"""
return self.compiled_model_.predict(X, batch_size=batch_size, verbose=verbose)
return self.compiled_model_.predict(X, batch_size=batch_size, verbose=verbose).ravel()
def score(self, X, y, batch_size=128, verbose=0):
"""

@ -82,16 +82,16 @@ print(classifier.compiled_model_)
print('Beginning scikit-learn regression wrapper test')
print('Generating data')
X_train = np.random((5000, 100))
X_test = np.random((1000, 100))
y_train = np.random(5000)
y_test = np.random(1000)
X_train = np.random.random((5000, 100))
X_test = np.random.random((1000, 100))
y_train = np.random.random(5000)
y_test = np.random.random(1000)
print('Defining model')
model = Sequential()
model.add(Dense(100, 50))
model.add(Activation('relu'))
model.add(Dense(50, 10))
model.add(Dense(50, 1))
model.add(Activation('linear'))
print('Creating wrapper')