Package ai.dat.reranker.onnx

Class OnnxScoringModel

java.lang.Object

ai.dat.reranker.onnx.OnnxScoringModel

All Implemented Interfaces:

dev.langchain4j.model.scoring.ScoringModel

public class OnnxScoringModel
extends Object

Constructor Summary

Constructors

Constructor	Description
OnnxScoringModel(String modelFileName, ai.onnxruntime.OrtSession.SessionOptions options, String tokenizerFileName)
OnnxScoringModel(String modelFileName, ai.onnxruntime.OrtSession.SessionOptions options, String tokenizerFileName, int modelMaxLength, boolean normalize)
OnnxScoringModel(String modelFileName, String tokenizerFileName)
OnnxScoringModel(String modelFileName, String tokenizerFileName, int modelMaxLength)

Method Summary

Modifier and Type	Method	Description
protected ai.dat.reranker.onnx.OnnxScoringBertCrossEncoder	model()
dev.langchain4j.model.output.Response<List<Double>>	scoreAll(List<dev.langchain4j.data.segment.TextSegment> segments, String query)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface dev.langchain4j.model.scoring.ScoringModel

score, score

Constructor Details

OnnxScoringModel

public OnnxScoringModel(String modelFileName,
 String tokenizerFileName)

OnnxScoringModel

public OnnxScoringModel(String modelFileName,
 ai.onnxruntime.OrtSession.SessionOptions options,
 String tokenizerFileName)

OnnxScoringModel

public OnnxScoringModel(String modelFileName,
 String tokenizerFileName,
 int modelMaxLength)

OnnxScoringModel

public OnnxScoringModel(String modelFileName,
 ai.onnxruntime.OrtSession.SessionOptions options,
 String tokenizerFileName,
 int modelMaxLength,
 boolean normalize)

Method Details

model

protected ai.dat.reranker.onnx.OnnxScoringBertCrossEncoder model()

scoreAll

public dev.langchain4j.model.output.Response<List<Double>> scoreAll(List<dev.langchain4j.data.segment.TextSegment> segments,
 String query)

Specified by:

scoreAll in interface dev.langchain4j.model.scoring.ScoringModel