3 overrides of GetPredictions
Microsoft.ML.TorchSharp (3)
NasBert\NerTrainer.cs (1)
206
private protected override torch.Tensor
GetPredictions
(torch.Tensor logits)
NasBert\SentenceSimilarityTrainer.cs (1)
140
private protected override torch.Tensor
GetPredictions
(torch.Tensor logits)
NasBert\TextClassificationTrainer.cs (1)
141
private protected override torch.Tensor
GetPredictions
(torch.Tensor logits)
1 reference to GetPredictions
Microsoft.ML.TorchSharp (1)
NasBert\NasBertTrainer.cs (1)
233
var predictions =
GetPredictions
(logits);