Skip to content

Commit

Permalink
Merge pull request #74 from amaiya/develop
Browse files Browse the repository at this point in the history
minor edit
  • Loading branch information
amaiya committed Mar 4, 2020
2 parents 3b91603 + b933e7b commit 33f9378
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion tutorials/tutorial-03-image-classification.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -825,6 +825,18 @@
"predictor.predict_folder(DATADIR+'/valid/cats/')[:10]"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"By default, `predict*` methods in *ktrain* return the predicted class labels. To view the predicted probabilities for each class, supply `return_proba=True` as an extra argument:\n",
"\n",
"```python\n",
"predictor.predict_filename(filename, return_proba=True)\n",
"predictor.predict_folder(foldername, return_proba=True)\n",
"```"
]
},
{
"cell_type": "markdown",
"metadata": {},
Expand Down Expand Up @@ -1545,7 +1557,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.6.8"
"version": "3.6.9"
}
},
"nbformat": 4,
Expand Down

0 comments on commit 33f9378

Please sign in to comment.