Example usage for org.apache.lucene.search.similarities DFRSimilarity getBasicModel

Introduction

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

Prototype

public BasicModel getBasicModel() 

Source Link

Document

Returns the basic model of information content

Usage

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

License:Apache License

/** dfr with default parameters */
public void test() throws Exception {
    Similarity sim = getSimilarity("text");
    assertEquals(DFRSimilarity.class, sim.getClass());
    DFRSimilarity dfr = (DFRSimilarity) sim;
    assertEquals(BasicModelIF.class, dfr.getBasicModel().getClass());
    assertEquals(AfterEffectB.class, dfr.getAfterEffect().getClass());
    assertEquals(NormalizationH2.class, dfr.getNormalization().getClass());
}

---

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

License:Apache License

/** dfr with parameterized normalization */
public void testParameters() throws Exception {
    Similarity sim = getSimilarity("text_params");
    assertEquals(DFRSimilarity.class, sim.getClass());
    DFRSimilarity dfr = (DFRSimilarity) sim;
    assertEquals(BasicModelIF.class, dfr.getBasicModel().getClass());
    assertEquals(AfterEffectB.class, dfr.getAfterEffect().getClass());
    assertEquals(NormalizationH3.class, dfr.getNormalization().getClass());
    NormalizationH3 norm = (NormalizationH3) dfr.getNormalization();
    assertEquals(900f, norm.getMu(), 0.01f);
}

---

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

License:Apache License

/** LUCENE-3566 */
public void testParameterC() throws Exception {
    Similarity sim = getSimilarity("text_paramc");
    assertEquals(DFRSimilarity.class, sim.getClass());
    DFRSimilarity dfr = (DFRSimilarity) sim;
    assertEquals(BasicModelP.class, dfr.getBasicModel().getClass());
    assertEquals(AfterEffectL.class, dfr.getAfterEffect().getClass());
    assertEquals(NormalizationH2.class, dfr.getNormalization().getClass());
    NormalizationH2 norm = (NormalizationH2) dfr.getNormalization();
    assertEquals(7f, norm.getC(), 0.01f);
}

---

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

License:Apache License

@Test
public void testResolveSimilaritiesFromMapping_DFR() throws IOException {
    String mapping = XContentFactory.jsonBuilder().startObject().startObject("type").startObject("properties")
            .startObject("field1").field("type", "string").field("similarity", "my_similarity").endObject()
            .endObject().endObject().endObject().string();

    Settings indexSettings = ImmutableSettings.settingsBuilder()
            .put("index.similarity.my_similarity.type", "DFR")
            .put("index.similarity.my_similarity.basic_model", "g")
            .put("index.similarity.my_similarity.after_effect", "l")
            .put("index.similarity.my_similarity.normalization", "h2")
            .put("index.similarity.my_similarity.normalization.h2.c", 3f).build();
    SimilarityService similarityService = similarityService(indexSettings);
    DocumentMapper documentMapper = similarityService.mapperService().documentMapperParser().parse(mapping);
    assertThat(documentMapper.mappers().name("field1").mapper().similarity(),
            instanceOf(DFRSimilarityProvider.class));

    DFRSimilarity similarity = (DFRSimilarity) documentMapper.mappers().name("field1").mapper().similarity()
            .get();//from w w w.j  a v a2  s .c o m
    assertThat(similarity.getBasicModel(), instanceOf(BasicModelG.class));
    assertThat(similarity.getAfterEffect(), instanceOf(AfterEffectL.class));
    assertThat(similarity.getNormalization(), instanceOf(NormalizationH2.class));
    assertThat(((NormalizationH2) similarity.getNormalization()).getC(), equalTo(3f));
}

---