Skip to content

Commit

Permalink
Update task.py
Browse files Browse the repository at this point in the history
  • Loading branch information
haifeng-jin committed Oct 1, 2019
1 parent c21dc7e commit 6bcad91
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions autokeras/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -256,7 +256,9 @@ def predict(self, x, batch_size=32, **kwargs):
"""Predict the output for a given testing data.
# Arguments
x: Any allowed types according to the input node. Testing data.
x: String, numpy.ndarray, pandas.DataFrame or tensorflow.Dataset.
Testing data x. If the data is from a csv file, it should be a
string specifying the path of the csv file of the testing data.
batch_size: Int. Defaults to 32.
**kwargs: Any arguments supported by keras.Model.predict.
Expand All @@ -275,9 +277,12 @@ def evaluate(self, x, y=None, batch_size=32, **kwargs):
"""Evaluate the best model for the given data.
# Arguments
x: Any allowed types according to the input node. Testing data.
y: Any allowed types according to the head. Testing targets.
Defaults to None.
x: String, numpy.ndarray, pandas.DataFrame or tensorflow.Dataset.
Testing data x. If the data is from a csv file, it should be a
string specifying the path of the csv file of the testing data.
y: String, numpy.ndarray, or tensorflow.Dataset. Testing data y.
If the data is from a csv file, it should be a string corresponding
to the label column.
batch_size: Int. Defaults to 32.
**kwargs: Any arguments supported by keras.Model.evaluate.
Expand Down

0 comments on commit 6bcad91

Please sign in to comment.