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