Update README.md
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.
|