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

Go to download
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;
  }
}