Skip to content

Commit 81401e9

Browse files
authored
Rename LinearClassificationTrainer to StochasticDualCoordinateAscent
1 parent 511503b commit 81401e9

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

src/Microsoft.ML.StandardLearners/Standard/LinearClassificationTrainer.cs

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@
2020
using System.Threading;
2121
using System.Threading.Tasks;
2222

23-
[assembly: LoadableClass(typeof(LinearClassificationTrainer), typeof(LinearClassificationTrainer.Arguments),
23+
[assembly: LoadableClass(typeof(StochasticDualCoordinateAscent), typeof(StochasticDualCoordinateAscent.Arguments),
2424
new[] { typeof(SignatureBinaryClassifierTrainer), typeof(SignatureTrainer), typeof(SignatureFeatureScorerTrainer) },
25-
LinearClassificationTrainer.UserNameValue,
26-
LinearClassificationTrainer.LoadNameValue,
25+
StochasticDualCoordinateAscent.UserNameValue,
26+
StochasticDualCoordinateAscent.LoadNameValue,
2727
"LinearClassifier",
2828
"lc",
2929
"sasdca")]
@@ -1361,7 +1361,7 @@ public void Add(Double summand)
13611361
}
13621362
}
13631363

1364-
public sealed class LinearClassificationTrainer : SdcaTrainerBase<BinaryPredictionTransformer<TScalarPredictor>, TScalarPredictor>
1364+
public sealed class StochasticDualCoordinateAscent : SdcaTrainerBase<BinaryPredictionTransformer<TScalarPredictor>, TScalarPredictor>
13651365
{
13661366
public const string LoadNameValue = "SDCA";
13671367
internal const string UserNameValue = "Fast Linear (SA-SDCA)";
@@ -1401,7 +1401,7 @@ internal override void Check(IHostEnvironment env)
14011401

14021402
public override TrainerInfo Info { get; }
14031403

1404-
public LinearClassificationTrainer(IHostEnvironment env, Arguments args,
1404+
public StochasticDualCoordinateAscent(IHostEnvironment env, Arguments args,
14051405
string featureColumn, string labelColumn, string weightColumn = null)
14061406
: base(Contracts.CheckRef(env, nameof(env)).Register(LoadNameValue), args, MakeFeatureColumn(featureColumn), MakeLabelColumn(labelColumn), MakeWeightColumn(weightColumn))
14071407
{
@@ -1431,7 +1431,7 @@ public LinearClassificationTrainer(IHostEnvironment env, Arguments args,
14311431

14321432
}
14331433

1434-
public LinearClassificationTrainer(IHostEnvironment env, Arguments args)
1434+
public StochasticDualCoordinateAscent(IHostEnvironment env, Arguments args)
14351435
: this(env, args, args.FeatureColumn, args.LabelColumn)
14361436
{
14371437
}
@@ -1903,19 +1903,19 @@ public static partial class Sdca
19031903
{
19041904
[TlcModule.EntryPoint(Name = "Trainers.StochasticDualCoordinateAscentBinaryClassifier",
19051905
Desc = "Train an SDCA binary model.",
1906-
UserName = LinearClassificationTrainer.UserNameValue,
1907-
ShortName = LinearClassificationTrainer.LoadNameValue,
1906+
UserName = StochasticDualCoordinateAscent.UserNameValue,
1907+
ShortName = StochasticDualCoordinateAscent.LoadNameValue,
19081908
XmlInclude = new[] { @"<include file='../Microsoft.ML.StandardLearners/Standard/doc.xml' path='doc/members/member[@name=""SDCA""]/*' />",
19091909
@"<include file='../Microsoft.ML.StandardLearners/Standard/doc.xml' path='doc/members/example[@name=""StochasticDualCoordinateAscentBinaryClassifier""]/*'/>" })]
1910-
public static CommonOutputs.BinaryClassificationOutput TrainBinary(IHostEnvironment env, LinearClassificationTrainer.Arguments input)
1910+
public static CommonOutputs.BinaryClassificationOutput TrainBinary(IHostEnvironment env, StochasticDualCoordinateAscent.Arguments input)
19111911
{
19121912
Contracts.CheckValue(env, nameof(env));
19131913
var host = env.Register("TrainSDCA");
19141914
host.CheckValue(input, nameof(input));
19151915
EntryPointUtils.CheckInputArgs(host, input);
19161916

1917-
return LearnerEntryPointsUtils.Train<LinearClassificationTrainer.Arguments, CommonOutputs.BinaryClassificationOutput>(host, input,
1918-
() => new LinearClassificationTrainer(host, input),
1917+
return LearnerEntryPointsUtils.Train<StochasticDualCoordinateAscent.Arguments, CommonOutputs.BinaryClassificationOutput>(host, input,
1918+
() => new StochasticDualCoordinateAscent(host, input),
19191919
() => LearnerEntryPointsUtils.FindColumn(host, input.TrainingData.Schema, input.LabelColumn),
19201920
calibrator: input.Calibrator, maxCalibrationExamples: input.MaxCalibrationExamples);
19211921
}

0 commit comments

Comments
 (0)