You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
356 B
Markdown
27 lines
356 B
Markdown
3 years ago
|
# sklearn example
|
||
|
|
||
|
This is a simple demo showing the training of Randmon Forest Classifier, serializing it as a pickle file, converting it to an Onnx model, then making inferences using the Onnx model.
|
||
|
|
||
|
## Usage
|
||
|
|
||
|
Train the model
|
||
|
|
||
|
```
|
||
|
python train.py
|
||
|
```
|
||
|
|
||
|
Convert to Onnx
|
||
|
|
||
|
```
|
||
|
|
||
|
```
|
||
|
|
||
|
Make inferences
|
||
|
|
||
|
```
|
||
|
|
||
|
```
|
||
|
|
||
|
## Model
|
||
|
|
||
|
![Model Graph](./img/model_graph.png)
|