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