Skip to content

Commit 27dd4e2

Browse files
committed
small fix
1 parent caa873b commit 27dd4e2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mirapy/classifiers/models.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -148,14 +148,14 @@ def train(self, x_train, y_train, epochs=50, batch_size=100,
148148
class_weight=class_weight, shuffle=True,
149149
verbose=verbose)
150150

151-
def predict(self, x_test):
151+
def predict(self, x):
152152
"""
153153
Predicts the output of the model for the given data as input.
154154
155-
:param x_test: Input data as Numpy arrays.
155+
:param x: Input data as Numpy arrays.
156156
:return: Predicted class for Input data.
157157
"""
158-
return self.model.predict_classes(x_test)
158+
return self.model.predict_classes(x)
159159

160160

161161
class AtlasVarStarClassifier(Classifier):

0 commit comments

Comments
 (0)