List of usage examples for org.apache.lucene.misc SweetSpotSimilarity SweetSpotSimilarity
public SweetSpotSimilarity()
From source file:com.o19s.es.ltr.query.LtrQueryTests.java
License:Apache License
@Before public void setupIndex() throws IOException { dirUnderTest = newDirectory();/*from ww w .j a va2s . c om*/ 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.eu.bitzone.Leia.java
License:Apache License
public Similarity createSimilarity(final Object srchOpts) { final Object ckSimDef = find(srchOpts, "ckSimDef"); final Object ckSimSweet = find(srchOpts, "ckSimSweet"); final Object ckSimOther = find(srchOpts, "ckSimOther"); final Object simClass = find(srchOpts, "simClass"); final Object ckSimCust = find(srchOpts, "ckSimCust"); if (getBoolean(ckSimDef, "selected")) { return new DefaultSimilarity(); } else if (getBoolean(ckSimSweet, "selected")) { return new SweetSpotSimilarity(); } else if (getBoolean(ckSimOther, "selected")) { try {//from w w w.j a v a 2 s. co m final Class clazz = Class.forName(getString(simClass, "text")); if (Similarity.class.isAssignableFrom(clazz)) { final Similarity sim = (Similarity) clazz.newInstance(); return sim; } else { throw new Exception("Not a subclass of Similarity: " + clazz.getName()); } } catch (final Exception e) { e.printStackTrace(); showStatus("ERROR: invalid Similarity, using default"); setBoolean(ckSimDef, "selected", true); setBoolean(ckSimOther, "selected", false); return new DefaultSimilarity(); } } else if (getBoolean(ckSimCust, "selected")) { return similarity; } else { return new DefaultSimilarity(); } }
From source file:org.getopt.luke.Luke.java
License:Apache License
public Similarity createSimilarity(Object srchOpts) { Object ckSimDef = find(srchOpts, "ckSimDef"); Object ckSimSweet = find(srchOpts, "ckSimSweet"); Object ckSimOther = find(srchOpts, "ckSimOther"); Object simClass = find(srchOpts, "simClass"); Object ckSimCust = find(srchOpts, "ckSimCust"); if (getBoolean(ckSimDef, "selected")) { return new DefaultSimilarity(); } else if (getBoolean(ckSimSweet, "selected")) { return new SweetSpotSimilarity(); } else if (getBoolean(ckSimOther, "selected")) { try {//from w w w. j a v a 2 s. co m Class clazz = Class.forName(getString(simClass, "text")); if (Similarity.class.isAssignableFrom(clazz)) { Similarity sim = (Similarity) clazz.newInstance(); return sim; } else { throw new Exception("Not a subclass of Similarity: " + clazz.getName()); } } catch (Exception e) { e.printStackTrace(); showStatus("ERROR: invalid Similarity, using default"); setBoolean(ckSimDef, "selected", true); setBoolean(ckSimOther, "selected", false); return new DefaultSimilarity(); } } else if (getBoolean(ckSimCust, "selected")) { return similarity; } else { return new DefaultSimilarity(); } }