studentscolab commited on
Commit
4830494
·
verified ·
1 Parent(s): c3f1937

Dodanie model.joblib + README

Browse files
Files changed (1) hide show
  1. README.md +29 -1
README.md CHANGED
@@ -18,4 +18,32 @@ Tabela z kolumnami:
18
 
19
  ## Wyjście
20
  - predict: klasa (0/1/2)
21
- - predict_proba: prawdopodobieństwa klas
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
  ## Wyjście
20
  - predict: klasa (0/1/2)
21
+ - predict_proba: prawdopodobieństwa klas
22
+
23
+ ## Przykład użycia
24
+ ```python
25
+ import joblib
26
+ import pandas as pd
27
+ from huggingface_hub import hf_hub_download
28
+ import numpy as np
29
+
30
+ repo_id = "studentscolab/iris"
31
+
32
+ model_path = hf_hub_download(repo_id=repo_id, filename="model.joblib")
33
+ model = joblib.load(model_path)
34
+
35
+ x = pd.DataFrame([{
36
+ "sepal length (cm)": 5.1,
37
+ "sepal width (cm)": 3.5,
38
+ "petal length (cm)": 1.4,
39
+ "petal width (cm)": 0.2,
40
+ }])
41
+
42
+ np.set_printoptions(precision=10, suppress=True)
43
+
44
+ pred = model.predict(x)[0]
45
+ proba = model.predict_proba(x)[0]
46
+ print("classes:", model.classes_)
47
+ print("pred:", pred)
48
+ print("proba:", proba)
49
+ ```