com.expleague.ml.data.set.impl.VecDataSetImpl 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.data.set.impl;

import com.expleague.commons.math.vectors.Mx;
import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.seq.Seq;
import com.expleague.ml.data.set.DataSet;
import com.expleague.ml.data.set.VecDataSet;
import com.expleague.ml.Vectorization;
import com.expleague.ml.meta.DSItem;

import java.util.stream.BaseStream;
import java.util.stream.IntStream;
import java.util.stream.Stream;

/**
 * User: solar
 * Date: 09.07.14
 * Time: 21:10
 */
public class VecDataSetImpl extends VecDataSet.Stub {
  private final Mx data;

  public VecDataSetImpl(final Mx data, final VecDataSet parent) {
    super(parent);
    this.data = data;
  }

  public  VecDataSetImpl(final DataSet parent, final Mx data, final Vectorization vec) {
    super(parent, vec);
    this.data = data;
  }

  @Override
  public Mx data() {
    return data;
  }

  @Override
  public Seq sub(int start, int end) {
    return new VecDataSetImpl(data.sub(start, 0, end - start, data.columns()), (VecDataSet)parent());
  }

  @Override
  public Stream stream() {
    return IntStream.range(0, length()).mapToObj(i -> data().row(i));
  }
}