Skip to content

Commit

Permalink
fix image indentation
Browse files Browse the repository at this point in the history
  • Loading branch information
sumn2u committed Jan 8, 2024
1 parent 4b6a58f commit f67a14e
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions paper/paper.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,13 +36,14 @@ Integration of machine learning models with mobile devices presents a promising

The model was trained with Tesla T4 GPU and uses EfficientNetV2 [@tan2021efficientnetv2] model as a base model with addition of agumentation layer. Adam was used as an optmizer with intital learning rate of 0.01. Which was later optmised using [optuna](https://optuna.org/) to create more accurate optimization parameters. The training and validation loss is shown in \autoref{fig:training_vs_val_loss} whereas \autoref{fig:training_vs_val_accuracy} shows training and validation accuracy on the performed experiment[^2].

![Training and Validation loss at different epochs\label{fig:training_vs_val_loss}](training_vs_val_loss.png){width="100%"}
![Training and Validation loss at different epochs\label{fig:training_vs_val_loss}](training_vs_val_loss.png){width="60%"}

![Training and Validation accuracy at different epochs\label{fig:training_vs_val_accuracy}](training_vs_val_accuracy.png){width="100%"}
![Training and Validation accuracy at different epochs\label{fig:training_vs_val_accuracy}](training_vs_val_accuracy.png){width="60%"}


The confusion matix of the modle is shown in \autoref{fig:confusion_matrix}.
![Confusion Matrix\label{fig:confusion_matrix}](confusion_matrix.png){width="100%"}
The confusion matix of the modle is shown in
\autoref{fig:confusion_matrix}.
![Confusion Matrix\label{fig:confusion_matrix}](confusion_matrix.png){width="60%"}

[^2]: [https://www.kaggle.com/code/sumn2u/garbage-classification-transfer-learning](https://www.kaggle.com/code/sumn2u/garbage-classification-transfer-learning).

Expand Down

0 comments on commit f67a14e

Please sign in to comment.