{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"from fastbook import *\n",
"from fastai.vision.all import *\n",
"from pathlib import Path"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"Path('data2')"
]
},
"execution_count": 2,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"path = Path('data2')\n",
"path"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {
"scrolled": false
},
"outputs": [
{
"data": {
"text/html": [
"
\n",
" \n",
" \n",
" epoch | \n",
" train_loss | \n",
" valid_loss | \n",
" error_rate | \n",
" time | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" 1.220044 | \n",
" 0.910795 | \n",
" 0.483412 | \n",
" 00:25 | \n",
"
\n",
" \n",
"
"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"\n",
" \n",
" \n",
" epoch | \n",
" train_loss | \n",
" valid_loss | \n",
" error_rate | \n",
" time | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" 1.037957 | \n",
" 0.840327 | \n",
" 0.487204 | \n",
" 00:31 | \n",
"
\n",
" \n",
" 1 | \n",
" 0.922677 | \n",
" 0.777105 | \n",
" 0.472986 | \n",
" 00:31 | \n",
"
\n",
" \n",
" 2 | \n",
" 0.822957 | \n",
" 0.745550 | \n",
" 0.496682 | \n",
" 00:31 | \n",
"
\n",
" \n",
" 3 | \n",
" 0.751167 | \n",
" 0.711318 | \n",
" 0.497630 | \n",
" 00:31 | \n",
"
\n",
" \n",
" 4 | \n",
" 0.720383 | \n",
" 0.703557 | \n",
" 0.472038 | \n",
" 00:31 | \n",
"
\n",
" \n",
"
"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"def is_buy(x): return x[0]==\"T\"\n",
"dls = ImageDataLoaders.from_name_func(\n",
" path, get_image_files(path), valid_pct=0.2, seed=42,\n",
" label_func=is_buy, item_tfms=Resize(224))\n",
"\n",
"learn = cnn_learner(dls, resnet34, metrics=error_rate)\n",
"learn.fine_tune(5)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.6"
}
},
"nbformat": 4,
"nbformat_minor": 4
}