com.expleague.ml.io.FMModelConversionPack Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmll

Various ML methods implemented by myself and my students

There is a newer version: 1.4.9

package com.expleague.ml.io;

import com.expleague.commons.func.converters.Vec2StringConverter;
import com.expleague.commons.func.types.ConversionPack;
import com.expleague.commons.func.types.TypeConverter;
import com.expleague.commons.math.io.Vec2CharSequenceConverter;
import com.expleague.commons.math.vectors.Mx;
import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.seq.CharSeqTools;
import com.expleague.ml.models.FMModel;
import com.expleague.commons.math.vectors.impl.mx.VecBasedMx;

/**
 * User: qdeee
 * Date: 31.03.14
 */
public class FMModelConversionPack implements ConversionPack {
  public static class To implements TypeConverter {
    @Override
    public CharSequence convert(final FMModel from) {
      final Vec2StringConverter vec2StringConverter = new Vec2StringConverter();
      final StringBuilder builder = new StringBuilder();
      builder.append(String.valueOf(from.getW0()));
      builder.append("\n");
      builder.append(vec2StringConverter.convertTo(from.getW()));
      builder.append("\n");
      builder.append(vec2StringConverter.convertTo(from.getV()));
      return builder;
    }
  }

  public static class From implements TypeConverter {
    @Override
    public FMModel convert(final CharSequence from) {
      final Vec2CharSequenceConverter vec2StringConverter = new Vec2CharSequenceConverter();
      final CharSequence[] lines = CharSeqTools.split(from, '\n');
      final double w0 = Double.valueOf(lines[0].toString());
      final Vec w = vec2StringConverter.convertFrom(lines[1].toString());
      final Vec vecV = vec2StringConverter.convertFrom(lines[2].toString());
      final Mx V = new VecBasedMx(w.dim(), vecV);
      return new FMModel(V, w, w0);
    }
  }


  @Override
  public Class to() {
    return To.class;
  }

  @Override
  public Class from() {
    return From.class;
  }
}