Resolved a few minor issues found during testing.
This commit is contained in:
parent
7c966439fa
commit
4afb5b60d6
@ -234,7 +234,7 @@ class KerasRegressor(BaseWrapper):
|
|||||||
preds : array-like, shape = (n_samples)
|
preds : array-like, shape = (n_samples)
|
||||||
Predictions.
|
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):
|
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('Beginning scikit-learn regression wrapper test')
|
||||||
|
|
||||||
print('Generating data')
|
print('Generating data')
|
||||||
X_train = np.random((5000, 100))
|
X_train = np.random.random((5000, 100))
|
||||||
X_test = np.random((1000, 100))
|
X_test = np.random.random((1000, 100))
|
||||||
y_train = np.random(5000)
|
y_train = np.random.random(5000)
|
||||||
y_test = np.random(1000)
|
y_test = np.random.random(1000)
|
||||||
|
|
||||||
print('Defining model')
|
print('Defining model')
|
||||||
model = Sequential()
|
model = Sequential()
|
||||||
model.add(Dense(100, 50))
|
model.add(Dense(100, 50))
|
||||||
model.add(Activation('relu'))
|
model.add(Activation('relu'))
|
||||||
model.add(Dense(50, 10))
|
model.add(Dense(50, 1))
|
||||||
model.add(Activation('linear'))
|
model.add(Activation('linear'))
|
||||||
|
|
||||||
print('Creating wrapper')
|
print('Creating wrapper')
|
||||||
|
Loading…
Reference in New Issue
Block a user