com.davidbracewell.apollo.mallet.MalletLabelEncoder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of apollo-extras Show documentation

A machine learning library for Java.

The newest version!

package com.davidbracewell.apollo.mallet;

import cc.mallet.types.Alphabet;
import com.davidbracewell.apollo.ml.Example;
import com.davidbracewell.apollo.ml.encoder.LabelEncoder;
import com.davidbracewell.apollo.ml.data.Dataset;
import com.davidbracewell.stream.MStream;

import java.io.Serializable;
import java.util.Arrays;
import java.util.List;

/**
 * @author David B. Bracewell
 */
public class MalletLabelEncoder implements LabelEncoder, Serializable {
   private static final long serialVersionUID = 1L;
   private final Alphabet labelAlphabet;

   public MalletLabelEncoder(Alphabet labelAlphabet) {
      this.labelAlphabet = labelAlphabet;
   }

   @Override
   public LabelEncoder createNew() {
      return new MalletLabelEncoder(labelAlphabet);
   }

   @Override
   public Object decode(double value) {
      return labelAlphabet.lookupObject((int) value);
   }

   @Override
   public double encode(Object object) {
      return labelAlphabet.lookupIndex(object);
   }

   @Override
   public void fit(Dataset dataset) {

   }

   @Override
   public void fit(MStream stream) {

   }

   @Override
   public void freeze() {
      labelAlphabet.stopGrowth();
   }

   @Override
   public double get(Object object) {
      return labelAlphabet.lookupIndex(object);
   }

   @Override
   public boolean isFrozen() {
      return labelAlphabet.growthStopped();
   }

   @Override
   public int size() {
      return labelAlphabet.size();
   }

   @Override
   public void unFreeze() {
      labelAlphabet.startGrowth();
   }

   @Override
   public List