[ADD] comments

This commit is contained in:
Sem van der Hoeven
2021-06-09 15:58:08 +02:00
parent 91744bc874
commit 365b2b2703

View File

@@ -2627,6 +2627,13 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"\"\"\"\n",
"compile het model met de binary crossentropy loss function. Dit doen we omdat\n",
"we maar 2 verschillende klassen hebben.\n",
"We gebruiken rmsprop als optimizer omdat INVULLEN\n",
"We gebruiken de accuracy als metric omdat we daarop willen trainen.\n",
"\"\"\"\n",
"\n",
"model.compile(loss='binary_crossentropy',\n", "model.compile(loss='binary_crossentropy',\n",
" optimizer='rmsprop',\n", " optimizer='rmsprop',\n",
" metrics=['accuracy'])" " metrics=['accuracy'])"
@@ -2792,6 +2799,9 @@
} }
], ],
"source": [ "source": [
"\n",
"#fit het model met de datasets. We gebruiken 70 epochs en 3 stappen per epoch.\n",
"\n",
"model_ft = model.fit(train_dataset,\n", "model_ft = model.fit(train_dataset,\n",
" steps_per_epoch=3,\n", " steps_per_epoch=3,\n",
" epochs=70,\n", " epochs=70,\n",
@@ -3188,6 +3198,7 @@
} }
], ],
"source": [ "source": [
"# laat het netwerk een prediction doen op alle test fotos.\n",
"dir_path = '../data/imgs/test/'\n", "dir_path = '../data/imgs/test/'\n",
"\n", "\n",
"for i in os.listdir(dir_path):\n", "for i in os.listdir(dir_path):\n",
@@ -3210,12 +3221,13 @@
] ]
}, },
{ {
"cell_type": "code", "cell_type": "markdown",
"execution_count": null, "id": "069341a9",
"id": "530bae52",
"metadata": {}, "metadata": {},
"outputs": [], "source": [
"source": [] "### Conclusie\n",
"We hebben het CNN getest en "
]
} }
], ],
"metadata": { "metadata": {