1 type derived from TorchSharpBaseTransformer
Microsoft.ML.TorchSharp (1)
NasBert\NasBertTrainer.cs (1)
411public abstract class NasBertTransformer<TLabelCol, TTargetsCol> : TorchSharpBaseTransformer<TLabelCol, TTargetsCol>
15 references to TorchSharpBaseTransformer
Microsoft.ML.TorchSharp (15)
NasBert\NasBertTrainer.cs (1)
500public NasBertMapper(TorchSharpBaseTransformer<TLabelCol, TTargetsCol> parent, DataViewSchema inputSchema) :
NasBert\NerTrainer.cs (3)
109private protected override TorchSharpBaseTransformer<VBuffer<uint>, TargetType> CreateTransformer(IHost host, Options options, torch.nn.Module model, DataViewSchema.DetachedColumn labelColumn) 262private protected override IRowMapper GetRowMapper(TorchSharpBaseTransformer<VBuffer<uint>, TargetType> parent, DataViewSchema schema) 366public Mapper(TorchSharpBaseTransformer<VBuffer<uint>, TargetType> parent, DataViewSchema inputSchema) : base(parent, inputSchema)
NasBert\SentenceSimilarityTrainer.cs (3)
107private protected override TorchSharpBaseTransformer<float, float> CreateTransformer(IHost host, Options options, torch.nn.Module model, DataViewSchema.DetachedColumn labelColumn) 193private protected override IRowMapper GetRowMapper(TorchSharpBaseTransformer<float, float> parent, DataViewSchema schema) 259public Mapper(TorchSharpBaseTransformer<float, float> parent, DataViewSchema inputSchema) : base(parent, inputSchema)
NasBert\TextClassificationTrainer.cs (3)
107private protected override TorchSharpBaseTransformer<uint, long> CreateTransformer(IHost host, Options options, torch.nn.Module model, DataViewSchema.DetachedColumn labelColumn) 198private protected override IRowMapper GetRowMapper(TorchSharpBaseTransformer<uint, long> parent, DataViewSchema schema) 302public Mapper(TorchSharpBaseTransformer<uint, long> parent, DataViewSchema inputSchema) : base(parent, inputSchema)
TorchSharpBaseTrainer.cs (5)
101TorchSharpBaseTransformer<TLabelCol, TTargetsCol> transformer = default; 129private protected abstract TorchSharpBaseTransformer<TLabelCol, TTargetsCol> CreateTransformer(IHost host, TorchSharpBaseTrainer<TLabelCol, TTargetsCol>.Options options, Module model, DataViewSchema.DetachedColumn labelColumn); 438private protected abstract IRowMapper GetRowMapper(TorchSharpBaseTransformer<TLabelCol, TTargetsCol> parent, DataViewSchema schema); 444private protected readonly TorchSharpBaseTransformer<TLabelCol, TTargetsCol> Parent; 455public TorchSharpBaseMapper(TorchSharpBaseTransformer<TLabelCol, TTargetsCol> parent, DataViewSchema inputSchema) :