Example usage for org.apache.lucene.search.similarities DFISimilarity DFISimilarity

List of usage examples for org.apache.lucene.search.similarities DFISimilarity DFISimilarity

Introduction

In this page you can find the example usage for org.apache.lucene.search.similarities DFISimilarity DFISimilarity.

Prototype

public DFISimilarity(Independence independenceMeasure) 

Source Link

Document

Create DFI with the specified divergence from independence measure

Usage

From source file:com.o19s.es.ltr.query.LtrQueryTests.java

License:Apache License

@Before
public void setupIndex() throws IOException {
    dirUnderTest = newDirectory();/*w  w  w.  j av  a2 s  .  co m*/
    List<Similarity> sims = Arrays.asList(new ClassicSimilarity(), new SweetSpotSimilarity(), // extends Classic
            new BM25Similarity(), new LMDirichletSimilarity(), new BooleanSimilarity(),
            new LMJelinekMercerSimilarity(0.2F), new AxiomaticF3LOG(0.5F, 10),
            new DFISimilarity(new IndependenceChiSquared()),
            new DFRSimilarity(new BasicModelBE(), new AfterEffectB(), new NormalizationH1()),
            new IBSimilarity(new DistributionLL(), new LambdaDF(), new NormalizationH3()));
    similarity = sims.get(random().nextInt(sims.size()));

    indexWriterUnderTest = new RandomIndexWriter(random(), dirUnderTest,
            newIndexWriterConfig().setSimilarity(similarity));
    for (int i = 0; i < docs.length; i++) {
        Document doc = new Document();
        doc.add(newStringField("id", "" + i, Field.Store.YES));
        doc.add(newField("field", docs[i], Store.YES));
        indexWriterUnderTest.addDocument(doc);
    }
    indexWriterUnderTest.commit();
    indexWriterUnderTest.forceMerge(1);
    indexWriterUnderTest.flush();

    indexReaderUnderTest = indexWriterUnderTest.getReader();
    searcherUnderTest = newSearcher(indexReaderUnderTest);
    searcherUnderTest.setSimilarity(similarity);
}

From source file:org.apache.solr.search.similarities.DFISimilarityFactory.java

License:Apache License

@Override
public Similarity getSimilarity() {
    DFISimilarity sim = new DFISimilarity(independenceMeasure);
    sim.setDiscountOverlaps(discountOverlaps);
    return sim;
}

From source file:org.elasticsearch.index.similarity.DFISimilarityProvider.java

License:Apache License

@Inject
public DFISimilarityProvider(@Assisted String name, @Assisted Settings settings) {
    super(name);//from ww w  .j  a v a 2s.c o  m
    boolean discountOverlaps = settings.getAsBoolean("discount_overlaps", true);
    Independence measure = parseIndependence(settings);
    this.similarity = new DFISimilarity(measure);
    this.similarity.setDiscountOverlaps(discountOverlaps);
}

From source file:org.elasticsearch.index.similarity.SimilarityProviders.java

License:Apache License

public static DFISimilarity createDfiSimilarity(Settings settings, Version indexCreatedVersion) {
    assertSettingsIsSubsetOf("DFI", indexCreatedVersion, settings, "independence_measure");

    return new DFISimilarity(parseIndependence(settings));
}