List of usage examples for org.apache.lucene.search.similarities DFISimilarity DFISimilarity
public DFISimilarity(Independence independenceMeasure)
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)); }