Geni will automatically detect whether the following optional dependencies are present:
[ml.dmlc/xgboost4j-spark_2.12 "1.0.0"]
[ml.dmlc/xgboost4j_2.12 "1.0.0"]
If so, the vars ml/xgboost-classifier
and ml/xgboost-regressor
will be bound to the appropriate functions. Otherwise, the vars will remain unbound.
An example of using XGBoost for classification:
(def xgb-model
(ml/fit
training
(ml/xgboost-classifier {:max-depth 2 :num-round 2})))
(-> dataframe
(ml/transform xgb-model)
(g/select :label :probability)
(g/limit 5)
g/show)
Can you improve this documentation?Edit on GitHub
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close