Proposal for a Clojure machine learning library.
Everything here is beta status. Breaking changes are unlikely but possible
Dependencies:
{:deps
{scicloj/scicloj.ml {:mvn/version "0.1.0-beta4"}}}
Code:
(require '[scicloj.ml.core :as ml]
'[scicloj.ml.metamorph :as mm]
'[scicloj.ml.dataset :as ds])
;; read train and test datasets
(def titanic-train
(->
(ds/dataset "https://github.com/scicloj/metamorph-examples/raw/main/data/titanic/train.csv"
{:key-fn keyword
:parser-fn :string})))
(def titanic-test
(->
(ds/dataset "https://github.com/scicloj/metamorph-examples/raw/main/data/titanic/test.csv"
{:key-fn keyword
:parser-fn :string})
(ds/add-column :Survived [""] :cycle)))
;; construct pipeline function including Logistic Regression model
(def pipe-fn
(ml/pipeline
(mm/select-columns [:Survived :Pclass ])
(mm/add-column :Survived (fn [ds] (map #(case % "1" "yes" "0" "no" nil "") (:Survived ds))))
(mm/categorical->number [:Survived :Pclass])
(mm/set-inference-target :Survived)
(mm/model {:model-type :smile.classification/logistic-regression})))
;; execute pipeline with train data including model in mode :fit
(def trained-ctx
(pipe-fn {:metamorph/data titanic-train
:metamorph/mode :fit}))
;; execute pipeline in mode :transform with test data which will do a prediction
(def test-ctx
(pipe-fn
(assoc trained-ctx
:metamorph/data titanic-test
:metamorph/mode :transform)))
;; extract prediction from pipeline function result
(-> test-ctx :metamorph/data
(ds/column-values->categorical :Survived))
;; => #tech.v3.dataset.column<string>[418]
;; :Survived
;; [no, no, yes, no, no, no, no, yes, no, no, no, no, no, yes, no, yes, yes, no, no, no...]
Full documentation is here:
API documentation: https://scicloj.github.io/scicloj.ml
This library itself is a shim, not containing any functions.
The code is present in the following repositories, and the functions get re-exported in scicloj.ml
in a
small number of namespaces for user convenience.
Scicloj.ml organises the existing code in 3 namespaces, as following:
Functions are re-exported from:
Functions are re-exported from:
Functions are re-exported from:
In case you are already familar with any of the original namespaces, they can of course be used directly as well:
(require '[tablecloth.api :as tc])
(tc/add-column ...)
Can you improve this documentation?Edit on GitHub
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close