Fill in model card
Browse files
README.md
CHANGED
@@ -1,3 +1,21 @@
|
|
1 |
---
|
2 |
license: mit
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: mit
|
3 |
---
|
4 |
+
|
5 |
+
## Garden Tutorial Model
|
6 |
+
|
7 |
+
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)
|
8 |
+
|
9 |
+
`model.joblib` is a serialized RandomForestClasifier trained with sklearn version 1.3.1. The training code is in Train_Model.ipynb. The model accepts a list of iris feature lists like
|
10 |
+
|
11 |
+
```
|
12 |
+
[
|
13 |
+
[5.5, 2.4, 3.7, 1. ],
|
14 |
+
[6.3, 2.8, 5.1, 1.5],
|
15 |
+
[6.4, 3.1, 5.5, 1.8],
|
16 |
+
[6.6, 3. , 4.4, 1.4],
|
17 |
+
...
|
18 |
+
]
|
19 |
+
```
|
20 |
+
|
21 |
+
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.
|