Ensemble.java

package org.wikibrain.sr.ensemble;

import gnu.trove.set.TIntSet;
import org.wikibrain.sr.SRResult;
import org.wikibrain.sr.SRResultList;

import java.io.IOException;
import java.util.List;

/**
 * @author Matt Lesicko
 */
public interface Ensemble {
    void trainSimilarity(List<EnsembleSim> simList);
    void trainMostSimilar(List<EnsembleSim> simList);
    SRResult predictSimilarity(List<SRResult> scores);
    SRResultList predictMostSimilar(List<SRResultList> scores, int maxResults, TIntSet validIds);
    void read(String file) throws IOException;
    void write(String file) throws IOException;
}