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

shorter, and more idomatic version of index-attr

Browse source
This commit is contained in:
Ben Mabey 2010-11-08 10:35:25 -07:00
parent 2d274d2598
commit 285e45717a
1 changed files with 5 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/clj_ml/data.clj
View file

@ -10,7 +10,8 @@
these data sets as well as their classes can be modified and assigned to
the instances. Finally data sets can be transformed into Clojure sequences
that can be transformed using usual Clojure functions like map, reduce, etc."
(:use [clj-ml utils])
(:use [clj-ml utils]
[clojure.contrib.seq :only [find-first]])
(:import (weka.core Instance Instances FastVector Attribute)
(cljml ClojureInstances)))
@ -38,15 +39,9 @@
(defn index-attr
"Returns the index of an attribute in the attributes definition of an
instance or dataset"
[dataset-or-instance attr]
(let [max (.numAttributes dataset-or-instance)
attrs (key-to-str attr)]
(loop [c 0]
(if (= c max)
(throw (.Exception (str "Attribute " attrs " not found")))
(if (= attrs (attribute-name-at dataset-or-instance c))
c
(recur (+ c 1 )))))))
[dataset attr-name]
(let [attr-name (name attr-name)]
(find-first #(= attr-name (.name (.attribute dataset %))) (range (.numAttributes dataset)))))
(defn attributes
"Returns the attributes (weka.core.Attribute) of the dataset or instance"