Kaggle dataset obtained from this link by Anubhab Swain.
Image obtained from Matthew Butler.
Models implemented by Anne Livia.
This dataset was compiled using a medical study conducted on a group of people.
This dataset shows a few variations of head sizes and masses of brains, it also consists additional gender and age group columns.
- Gender: 1 for Male, 2 for Female
- Age Range: 1 represents > 18 years of age, 2 represents < 18 years of age
- Head Size(cm^3): Head volume in cubic centimetres
- Brain Weight(grams): - Mass of brains in grams
Predict the masses of the brains using the data, and later compare it with the actual masses of brains mentioned in the dataset.
- Python
- Pandas
- Scikit-learn
- Matplotlib
- Seaborn
- Linear Regression:
- R2 (only Head Size(cm^3) considered) = 0.73
- R2 (All three attributes) = 0.74
- Multi-layer Perceptron:
- R2 (All three attributes) = 0.74
- Decision Tree:
- R2 (All three attributes) = 0.69