diff --git a/src/clj_ml/io.clj b/src/clj_ml/io.clj index 00d3590..31d5347 100644 --- a/src/clj_ml/io.clj +++ b/src/clj_ml/io.clj @@ -7,8 +7,8 @@ clj-ml.io "Functions for reading and saving datasets, classifiers and clusterers to files and other persistence mechanisms." - (:import (weka.core.converters CSVLoader ArffLoader XRFFLoader) - (weka.core.converters CSVSaver ArffSaver XRFFSaver) + (:import (weka.core.converters CSVLoader ArffLoader XRFFLoader LibSVMLoader) + (weka.core.converters CSVSaver ArffSaver XRFFSaver LibSVMSaver) (java.io File InputStream OutputStream) (java.net URL URI))) @@ -44,6 +44,11 @@ (let [loader (new CSVLoader)] (m-load-instances loader source)))) +(defmethod load-instances :libsvm + ([kind source & options] + (let [loader (new LibSVMLoader)] + (m-load-instances loader source)))) + ;; Saving of instances (defmulti save-instances @@ -73,3 +78,8 @@ ([kind destiny instances & options] (let [saver (new CSVSaver)] (m-save-instances saver destiny instances)))) + +(defmethod save-instances :libsvm + ([kind destiny instances & options] + (let [saver (new LibSVMSaver)] + (m-save-instances saver destiny instances))))