Skip to content

scKonung/Crab-Age-Regression

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Crab Age Prediction Model

This project marks my debut in Kaggle competitions, where I developed a straightforward model to predict the age of crabs.

Dataset Overview

The dataset I utilized for this model is sourced from this dataset. To enhance the model's input, I applied One Hot Encoding to the 'Sex' column, creating individual columns for each sex category.

Original Data Snapshot:
Original Data

Encoded Data Snapshot:
Encoded Data

Model Overview

The model architecture I employed consists of three Dense layers, with the first two layers having 32 output units each, and the final layer outputting a single value for age prediction. I utilized ReLU activation functions throughout.

Model Training and Optimization

For training optimization, I chose the RMSprop optimizer and Mean Squared Error (MSE) as the loss function. During the competition phase, I trained the model over 100 epochs with a batch size of 32.

This approach allowed me to create an effective baseline model for predicting crab ages, leveraging straightforward yet powerful techniques to tackle the problem at hand.

Releases

No releases published

Packages

No packages published