Implement predict and score methods & tests

2025-08-15 23:46:02 +00:00 · 2020-05-13 12:42:09 +02:00
parent c4de782a3f
commit 8f71eeb316
6 changed files with 70 additions and 221 deletions
--- a/main.py
+++ b/main.py
@@ -9,3 +9,4 @@ model = Stree(random_state=random_state)
 model.fit(X, y)
 print(model)
 model.save_sub_datasets()
+print(f"Prediciting [{y[0]}] we have {model.predict(X[0, :].reshape(-1, X.shape[1]))}")