diff --git a/Classification Sample Models/lymphoma-cancer-classification/model.ipynb b/Classification Sample Models/lymphoma-cancer-classification/model.ipynb index f1fa150..86e7aab 100644 --- a/Classification Sample Models/lymphoma-cancer-classification/model.ipynb +++ b/Classification Sample Models/lymphoma-cancer-classification/model.ipynb @@ -476,7 +476,7 @@ "import math\n", "\n", "# adam = optimizers.Adam(lr=0.001, beta_1=0.9, beta_2=0.999)\n", - "sgd = optimizers.SGD(lr=0.01)\n", + "sgd = optimizers.SGD(lr=0.02)\n", "\n", "filepath=\"./models/weights-{epoch:02d}-{val_acc:.2f}.hdf5\"\n", "\n", @@ -501,7 +501,7 @@ " steps_per_epoch=train_step,\n", " validation_data=validation_generator,\n", " validation_steps=valid_step,\n", - " epochs=20,\n", + " epochs=30,\n", " callbacks=callbacks_list,\n", " verbose=1)\n", "\n", @@ -524,4 +524,4 @@ "outputs": [] } ] -} \ No newline at end of file +}