2 types derived from TrainStateBase
Microsoft.ML.StandardTrainers (2)
Standard\Online\AveragedLinear.cs (1)
132private protected abstract class AveragedTrainStateBase : TrainStateBase
Standard\Online\LinearSvm.cs (1)
117private sealed class TrainState : TrainStateBase
8 references to TrainStateBase
Microsoft.ML.StandardTrainers (8)
Standard\Online\AveragedPerceptron.cs (1)
212private protected override TrainStateBase MakeState(IChannel ch, int numFeatures, LinearModelParameters predictor)
Standard\Online\LinearSvm.cs (1)
320private protected override TrainStateBase MakeState(IChannel ch, int numFeatures, LinearModelParameters predictor)
Standard\Online\OnlineGradientDescent.cs (1)
181private protected override TrainStateBase MakeState(IChannel ch, int numFeatures, LinearModelParameters predictor)
Standard\Online\OnlineLinear.cs (5)
201/// Called by <see cref="TrainCore(IChannel, RoleMappedData, TrainStateBase)"/> at the start of a pass over the dataset. 212/// Called by <see cref="TrainCore(IChannel, RoleMappedData, TrainStateBase)"/> after a pass over the dataset. 294var state = MakeState(ch, numFeatures, initLinearPred); 313private void TrainCore(IChannel ch, RoleMappedData data, TrainStateBase state) 351private protected abstract TrainStateBase MakeState(IChannel ch, int numFeatures, LinearModelParameters predictor);