Skip to content

Commit 0f75960

Browse files
authored
Update README.md
1 parent 6b249d9 commit 0f75960

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

README.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,12 +56,11 @@ which might be a better quick start experience, still full featured as same code
5656
;; the (single, fixed) pipe-fn
5757
(def pipe-fn
5858
(morph/pipeline
59+
;; set inference traget column
5960
(ds-mm/set-inference-target :species)
61+
;; convert all categorical variables to numbers
6062
(ds-mm/categorical->number cf/categorical)
61-
;; sets the ground truth for the prediction into the context at the required key
62-
(fn [ctx]
63-
(assoc ctx
64-
:scicloj.metamorph.ml/target-ds (cf/target (:metamorph/data ctx))))
63+
;; train or predict , depending on :mode
6564
{:metamorph/id :model}
6665
(ml/model {:model-type :smile.classification/random-forest})))
6766

0 commit comments

Comments
 (0)