Downloads · 30 days
0
SIH/penguin-classifier-sklearn
penguin-classifier-sklearn is a tabular classification model from SIH. Use it for the tabular classification task on the model card, and read the license before you ship it in a product. It is set up for sklearn. The card lists the license as mit.
This model is a scikit-learn classifier trained to predict the species of penguins in the Palmer Penguins dataset. The dataset contains measurements of penguin species including the species itself, making it suitable…
Downloads · 30 days
0
Access
Public
Updated Mar 11, 2024
Repo size
10.9 KB
Likes
0
Public
Click a slice to open those files.
.pkl10.9 KB · 74%
From the Hugging Face model README
This model is a scikit-learn classifier trained to predict the species of penguins in the Palmer Penguins dataset. The dataset contains measurements of penguin species including the species itself, making it suitable for classification tasks.
The model uses features such as culmen length, culmen depth, flipper length, and body mass to predict the species of penguins.
The model is intended for classifying the species of penguins based on their physical measurements. It can be used in applications related to penguin species classification and analysis.
Limitations:
The model is trained on the Palmer Penguins dataset, which contains measurements of penguin species including Adelie, Chinstrap, and Gentoo. The dataset is publicly available and can be accessed here.
The model is trained using scikit-learn, a popular machine learning library in Python. It uses a classification algorithm (e.g., Random Forest, Support Vector Machine) to learn the relationship between the input features (culmen length, culmen depth, flipper length, body mass) and the target variable (species).
# Load the model
from sklearn.externals import joblib
model = joblib.load('path/to/your/model.pkl')
# Input features
features = {
'culmen_length_mm': 39.1,
'culmen_depth_mm': 18.7,
'flipper_length_mm': 181,
'body_mass_g': 3750
}
# Predict species
predicted_species = model.predict([list(features.values())])[0]
print(f"Predicted species: {predicted_species}")
No penguins were harmed while training this model 🐧.
We were noot involved in collecting the 🐧 data.