Skip to content

Custom Deep Learning models developed for Image Classification and Time Series Prediction challenges.

Notifications You must be signed in to change notification settings

titaniumwhite/AN2DL-challenges

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Deep Learning challenges

@Politecnico di Milano, Academic Year 2021-2022, Artificial Neural Network and Deep Learning course

The aim of the two challenges was to develop multiples models of neural network in order to solve specific problems. This first challenge aimed to classify images of leaves using Convolutional Neural Networks or Transfer Learning. Instead, in the second challenge we had to create a Neural Network to predict future samples of a multivariate series.

Challenge1 - leaf challenge

Firstly, the dataset has been analysed. We applied data augmentation to enhance the model's ability to handle diverse leaf images. Additionally, class imbalance was addressed through class_weight from sklearn and the dataset was divided into training and validation sets. Keras TensorBoard callbacks were used to monitor and optimize the learning process. Among all the models, the one which gave best accuracy is model3 which makes use of Transfer Learning (VGG16) and Fine Tuning and a series of multiple Dropout and Dense Layers.









Challenge2 - timeseries challenge

In the second challenge we had to create a Neural Network to predict future samples of a multivariate series. Once divided the dataset in train and validation, we decided to use during the first training a window of 200 and a stride of 10. During the challenge, we fine-tuned these two parameters. The model which gave a best accuray is model4, composed by an LSTM of 64 units, a dropout layer, another LSTM of 128 units, one Global Average Pooling layer and another dropout. Both LSTMs had kernel regularized and recurrent regularizer.

About

Custom Deep Learning models developed for Image Classification and Time Series Prediction challenges.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published