Search code examples
javamachine-learningclassificationmallet

How to use array of doubles as feature vector in Mallet

From what I've seen in documentation and various examples, typical worfklow with data in Mallet requires you to work with feature list that you usually obtain by passing your data through "pipes" while iterating over them with some sort of iterator. The data is ususally stored in some csv file.

I am trying to obtain features list from two arrays of doubles. One array stores actual features and is of size n x m (where n is amount of features and m is count of feature vectors) and other one of size 1 x m and contains binary labels. How should I convert those into feature list, so I can use them in classifiers.

Solution

  • I ended up writing custom Itereator similar to the one present in Mullet called "ArrayDataAndTargetIterator". I also had to use a pipe defined like this:

    new SerialPipes(Arrays.asList(new Target2Label(), new Array2FeatureVector()));