class ZeroShotClassificationModel
extends Model
import { ZeroShotClassificationModel } from "https://deno.land/x/bertml@v0.1.0-alpha2/mod.ts";
A model for creating custom labels and checking to see if these labels can be matched on certain inputs.
Constructors
new
ZeroShotClassificationModel(manager: ModelManager, rid: number)Methods
predict(init: ZeroShotPredictInit): Promise<Label[]>
Predicts the label that has the highest confidence score for each of the inputs.
predictMultilabel(init: ZeroShotPredictInit): Promise<Label[][]>
Predicts the score of each label for each of the inputs.