All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.expleague.ml.data.tools.CatboostPool Maven / Gradle / Ivy

There is a newer version: 1.4.9
Show newest version
package com.expleague.ml.data.tools;

import com.expleague.commons.math.vectors.Mx;
import com.expleague.commons.seq.Seq;
import com.expleague.ml.data.tools.FakePool;

import java.util.Set;

/**
* User:  noxooomo
*/
public class CatboostPool extends FakePool {
  private final Set catColumns;

  public CatboostPool(final Mx data,
                      final Seq target,
                      final Set catColumns) {
    super(data, target, FakePool::genItems);
    this.catColumns = catColumns;
  }

  public boolean isCatFactor(int factorId) {
    return catColumns.contains(factorId);
  }

  public Set getCatFeatureIds() {
    return catColumns;
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy