yogsototh/clj-ml
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

additional options to weka models

Browse source
This commit is contained in:
Ronny Bjarnason 2010-12-14 15:04:42 -07:00
parent 44f63508c9
commit c89e559ace
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/clj_ml/classifiers.clj
View file

@ -121,6 +121,7 @@
(->> (check-options m {:fit-logistic-models "-M"})
(check-option-values m
{:complexity-constant "-C"
:normalize "-N"
:tolerance "-L"
:epsilon-roundoff "-P"
:folds-for-cross-validation "-V"
@ -165,14 +166,21 @@
(defmethod make-classifier-options [:decision-tree :random-forest]
([kind algorithm m]
(->> (check-option-values m
(->>
(check-options m {:debug "-D"})
(check-option-values m
{:num-trees-in-forest "-I"
:num-features-to-consider "-K"
:random-seed "-S"}))))
:random-seed "-S"
:depth "-depth"}))))
(defmethod make-classifier-options [:decision-tree :m5p]
([kind algorithm m]
(->> (check-options m {:unsmoothed-predictions "-U"}))))
(->>
(check-options m {:unsmoothed-predictions "-U"
:regression "-R"
:unpruned "-N"})
(check-option-values m {:minimum-instances "-M"}))))