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

adds tests for dataset label helper fns

Browse source
This commit is contained in:
Ben Mabey 2010-12-07 13:48:52 -07:00
parent 6a30d86841
commit 99048715f1
2 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/clj_ml/data.clj
View file

@ -184,8 +184,8 @@
(defn dataset-class-labels
"Returns the possible labels for the class attribute"
[dataset]
(let [class-attr (.classAttribute dataset)
[^Instances dataset]
(let [^Attribute class-attr (.classAttribute dataset)
values (.enumerateValues class-attr)]
(loop [continue (.hasMoreElements values)
acum {}]

6
test/clj_ml/data_test.clj
View file

@ -146,3 +146,9 @@
(let [ds (make-dataset "test" [:a {:b [:foo :bar]}] [[1 :foo] [2 :bar]])
_ (dataset-replace-attribute! ds :b (nominal-attribute :b [:baz :shaz]))]
(is (= [:a {:b [:shaz :baz]}] (dataset-format ds)))))
(deftest dataset-label-helpers
(let [ds (make-dataset "test" [:a :b {:c [:d :e]}]
[{:a 1 :b 2 :c :d} [4 5 :e]])]
(dataset-set-class ds :c)
(is (= {:d 0 :e 1} (dataset-class-labels ds) (dataset-labels-at ds 2)))))