sklearn-iris / README.md
hholb's picture
Update README.md
9e4dfc2 verified
|
raw
history blame
755 Bytes
---
license: mit
---
## Garden Tutorial Model
This repo contains a groundbreaking AI model that classifies species of irises by sepal length, sepal width, petal length and petal width. [See here for details on the training dataset.](https://archive.ics.uci.edu/dataset/53/iris)
`model.joblib` is a serialized RandomForestClasifier trained with sklearn version 1.4.1.post1. The training code is in Train_Model.ipynb. The model accepts a list of iris feature lists like
```
[
[5.5, 2.4, 3.7, 1. ],
[6.3, 2.8, 5.1, 1.5],
[6.4, 3.1, 5.5, 1.8],
[6.6, 3. , 4.4, 1.4],
...
]
```
where each list is `[sepal_length, sepal_width, petal_length, petal_width]`. The model outputs an iris category as an integer. 0 is setosa, 1 is versicolor, 2 is virginica.