Skip to content

Commit ea0915d

Browse files
Update feedforward.py (#16)
1 parent d9258b2 commit ea0915d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/feedforward.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def create_model(self, X=None, y=None):
1919
num_samples, self.num_out = Y.shape
2020

2121
model_input = theano.shared(np.zeros(shape=(1, self.num_pred)))
22-
model_output = theano.shared(np.zeros(self.num_out))
22+
model_output = theano.shared(np.zeros(shape=(1,self.num_out)))
2323

2424
self.shared_vars = {
2525
'model_input': model_input,

0 commit comments

Comments
 (0)