Skip to content

Commit acc45b9

Browse files
committed
private protected stuff, other comments
1 parent 543c81c commit acc45b9

File tree

18 files changed

+59
-57
lines changed

18 files changed

+59
-57
lines changed

src/Microsoft.ML.Core/Utilities/ObjectPool.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ public abstract class ObjectPoolBase<T>
3939
public int Count => _pool.Count;
4040
public int NumCreated { get { return _numCreated; } }
4141

42-
protected internal ObjectPoolBase()
42+
private protected ObjectPoolBase()
4343
{
4444
_pool = new ConcurrentBag<T>();
4545
}

src/Microsoft.ML.Data/Commands/TrainCommand.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,10 +235,10 @@ public static IPredictor Train(IHostEnvironment env, IChannel ch, RoleMappedData
235235
}
236236

237237
public static IPredictor Train(IHostEnvironment env, IChannel ch, RoleMappedData data, ITrainer trainer, string name, RoleMappedData validData,
238-
SubComponent<ICalibratorTrainer, SignatureCalibrator> calibrator, int maxCalibrationExamples, bool? cacheData, IPredictor inpPredictor = null)
238+
SubComponent<ICalibratorTrainer, SignatureCalibrator> calibrator, int maxCalibrationExamples, bool? cacheData, IPredictor inputPredictor = null)
239239
{
240240
ICalibratorTrainer caliTrainer = !calibrator.IsGood() ? null : calibrator.CreateInstance(env);
241-
return TrainCore(env, ch, data, trainer, name, validData, caliTrainer, maxCalibrationExamples, cacheData, inpPredictor);
241+
return TrainCore(env, ch, data, trainer, name, validData, caliTrainer, maxCalibrationExamples, cacheData, inputPredictor);
242242
}
243243

244244
private static IPredictor TrainCore(IHostEnvironment env, IChannel ch, RoleMappedData data, ITrainer trainer, string name, RoleMappedData validData,

src/Microsoft.ML.Data/Training/EarlyStoppingCriteria.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,9 +139,9 @@ public class Arguments : ArgumentsBase
139139
public int WindowSize = 5;
140140
}
141141

142-
protected internal Queue<Float> PastScores;
142+
protected Queue<Float> PastScores;
143143

144-
internal MovingWindowEarlyStoppingCriterion(Arguments args, bool lowerIsBetter)
144+
private protected MovingWindowEarlyStoppingCriterion(Arguments args, bool lowerIsBetter)
145145
: base(args, lowerIsBetter)
146146
{
147147
Contracts.CheckUserArg(0 <= Args.Threshold && args.Threshold <= 1, nameof(args.Threshold), "Must be in range [0,1].");

src/Microsoft.ML.Data/Training/TrainerBase.cs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,11 @@ namespace Microsoft.ML.Runtime.Training
77
public abstract class TrainerBase<TPredictor> : ITrainer<TPredictor>, ITrainerEx
88
where TPredictor : IPredictor
99
{
10-
public const string NoTrainingInstancesMessage = "No valid training instances found, all instances have missing features.";
10+
/// <summary>
11+
/// A standard string to use in errors or warnings by subclasses, to communicate the idea that no valid
12+
/// instances were able to be found.
13+
/// </summary>
14+
protected const string NoTrainingInstancesMessage = "No valid training instances found, all instances have missing features.";
1115

1216
protected IHost Host { get; }
1317

src/Microsoft.ML.Ensemble/Selector/SubModelSelector/BaseDiverseSelector.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ public abstract class DiverseSelectorArguments : ArgumentsBase
2323
private readonly IComponentFactory<IDiversityMeasure<TOutput>> _diversityMetricType;
2424
private ConcurrentDictionary<FeatureSubsetModel<IPredictorProducing<TOutput>>, TOutput[]> _predictions;
2525

26-
protected internal BaseDiverseSelector(IHostEnvironment env, DiverseSelectorArguments args, string name,
26+
private protected BaseDiverseSelector(IHostEnvironment env, DiverseSelectorArguments args, string name,
2727
IComponentFactory<IDiversityMeasure<TOutput>> diversityMetricType)
2828
: base(args, env, name)
2929
{

src/Microsoft.ML.Ensemble/Trainer/Binary/EnsembleTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ public EnsembleTrainer(IHostEnvironment env, Arguments args)
6262

6363
public override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
6464

65-
protected internal override TScalarPredictor CreatePredictor(List<FeatureSubsetModel<TScalarPredictor>> models)
65+
private protected override TScalarPredictor CreatePredictor(List<FeatureSubsetModel<TScalarPredictor>> models)
6666
{
6767
if (models.All(m => m.Predictor is TDistPredictor))
6868
return new EnsembleDistributionPredictor(Host, PredictionKind, CreateModels<TDistPredictor>(models), Combiner);

src/Microsoft.ML.Ensemble/Trainer/EnsembleTrainerBase.cs

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -59,17 +59,17 @@ public abstract class ArgumentsBase : LearnerInputBaseWithLabel
5959

6060
private const int DefaultNumModels = 50;
6161
/// <summary> Command-line arguments </summary>
62-
protected internal readonly ArgumentsBase Args;
63-
protected internal readonly int NumModels;
62+
private protected readonly ArgumentsBase Args;
63+
private protected readonly int NumModels;
6464

6565
/// <summary> Ensemble members </summary>
66-
protected internal readonly ITrainer<IPredictorProducing<TOutput>>[] Trainers;
66+
private protected readonly ITrainer<IPredictorProducing<TOutput>>[] Trainers;
6767

6868
private readonly ISubsetSelector _subsetSelector;
69-
protected internal ISubModelSelector<TOutput> SubModelSelector;
70-
protected internal IOutputCombiner<TOutput> Combiner;
69+
private protected ISubModelSelector<TOutput> SubModelSelector;
70+
private protected IOutputCombiner<TOutput> Combiner;
7171

72-
protected internal EnsembleTrainerBase(ArgumentsBase args, IHostEnvironment env, string name)
72+
private protected EnsembleTrainerBase(ArgumentsBase args, IHostEnvironment env, string name)
7373
: base(env, name)
7474
{
7575
Args = args;
@@ -133,7 +133,7 @@ private TPredictor TrainCore(IChannel ch, RoleMappedData data)
133133
validationDataSetProportion = Math.Max(validationDataSetProportion, stackingTrainer.ValidationDatasetProportion);
134134

135135
var needMetrics = Args.ShowMetrics || Combiner is IWeightedAverager;
136-
var Models = new List<FeatureSubsetModel<IPredictorProducing<TOutput>>>();
136+
var models = new List<FeatureSubsetModel<IPredictorProducing<TOutput>>>();
137137

138138
_subsetSelector.Initialize(data, NumModels, Args.BatchSize, validationDataSetProportion);
139139
int batchNumber = 1;
@@ -179,16 +179,16 @@ private TPredictor TrainCore(IChannel ch, RoleMappedData data)
179179
if (stackingTrainer != null)
180180
stackingTrainer.Train(modelsList, _subsetSelector.GetTestData(null, batch), Host);
181181

182-
Models.AddRange(modelsList);
183-
int modelSize = Utils.Size(Models);
182+
models.AddRange(modelsList);
183+
int modelSize = Utils.Size(models);
184184
if (modelSize < Utils.Size(Trainers))
185185
ch.Warning("{0} of {1} trainings failed.", Utils.Size(Trainers) - modelSize, Utils.Size(Trainers));
186186
ch.Check(modelSize > 0, "Ensemble training resulted in no valid models.");
187187
}
188-
return CreatePredictor(Models);
188+
return CreatePredictor(models);
189189
}
190190

191-
protected internal abstract TPredictor CreatePredictor(List<FeatureSubsetModel<IPredictorProducing<TOutput>>> models);
191+
private protected abstract TPredictor CreatePredictor(List<FeatureSubsetModel<IPredictorProducing<TOutput>>> models);
192192

193193
private bool EnsureMinimumFeaturesSelected(Subset subset)
194194
{
@@ -203,7 +203,7 @@ private bool EnsureMinimumFeaturesSelected(Subset subset)
203203
return false;
204204
}
205205

206-
protected internal virtual void PrintMetrics(IChannel ch, List<FeatureSubsetModel<IPredictorProducing<TOutput>>> models)
206+
private protected virtual void PrintMetrics(IChannel ch, List<FeatureSubsetModel<IPredictorProducing<TOutput>>> models)
207207
{
208208
// REVIEW: The formatting of this method is bizarre and seemingly not even self-consistent
209209
// w.r.t. its usage of |. Is this intentional?
@@ -216,7 +216,7 @@ protected internal virtual void PrintMetrics(IChannel ch, List<FeatureSubsetMode
216216
ch.Info("{0}{1}", string.Join("", model.Metrics.Select(m => string.Format("| {0} |", m.Value))), model.Predictor.GetType().Name);
217217
}
218218

219-
protected internal static FeatureSubsetModel<T>[] CreateModels<T>(List<FeatureSubsetModel<IPredictorProducing<TOutput>>> models) where T : IPredictor
219+
private protected static FeatureSubsetModel<T>[] CreateModels<T>(List<FeatureSubsetModel<IPredictorProducing<TOutput>>> models) where T : IPredictor
220220
{
221221
var subsetModels = new FeatureSubsetModel<T>[models.Count];
222222
for (int i = 0; i < models.Count; i++)

src/Microsoft.ML.Ensemble/Trainer/Multiclass/MulticlassDataPartitionEnsembleTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ public MulticlassDataPartitionEnsembleTrainer(IHostEnvironment env, Arguments ar
6363

6464
public override PredictionKind PredictionKind => PredictionKind.MultiClassClassification;
6565

66-
protected internal override EnsembleMultiClassPredictor CreatePredictor(List<FeatureSubsetModel<TVectorPredictor>> models)
66+
private protected override EnsembleMultiClassPredictor CreatePredictor(List<FeatureSubsetModel<TVectorPredictor>> models)
6767
{
6868
return new EnsembleMultiClassPredictor(Host, CreateModels<TVectorPredictor>(models), Combiner as IMultiClassOutputCombiner);
6969
}

src/Microsoft.ML.Ensemble/Trainer/Regression/RegressionEnsembleTrainer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ public RegressionEnsembleTrainer(IHostEnvironment env, Arguments args)
5757

5858
public override PredictionKind PredictionKind => PredictionKind.Regression;
5959

60-
protected internal override TScalarPredictor CreatePredictor(List<FeatureSubsetModel<TScalarPredictor>> models)
60+
private protected override TScalarPredictor CreatePredictor(List<FeatureSubsetModel<TScalarPredictor>> models)
6161
{
6262
return new EnsemblePredictor(Host, PredictionKind, CreateModels<TScalarPredictor>(models), Combiner);
6363
}

src/Microsoft.ML.FastTree/FastTree.cs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ public abstract class FastTreeTrainerBase<TArgs, TPredictor> :
8989

9090
public override bool SupportsValidation => true;
9191

92-
protected internal FastTreeTrainerBase(IHostEnvironment env, TArgs args)
92+
private protected FastTreeTrainerBase(IHostEnvironment env, TArgs args)
9393
: base(env, RegisterName)
9494
{
9595
Host.CheckValue(args, nameof(args));
@@ -1880,7 +1880,7 @@ private void MakeBoundariesAndCheckLabels(out long missingInstances, out long to
18801880
missingInstances = cursor.BadFeaturesRowCount;
18811881
}
18821882

1883-
ch.Check(totalInstances > 0, TrainerBase<IPredictor>.NoTrainingInstancesMessage);
1883+
ch.Check(totalInstances > 0, "All instances skipped due to missing features.");
18841884

18851885
if (missingInstances > 0)
18861886
ch.Warning("Skipped {0} instances with missing features during training", missingInstances);
@@ -2806,7 +2806,7 @@ public abstract class FastTreePredictionWrapper :
28062806
public bool CanSavePfa => true;
28072807
public bool CanSaveOnnx => true;
28082808

2809-
protected internal FastTreePredictionWrapper(IHostEnvironment env, string name, Ensemble trainedEnsemble, int numFeatures, string innerArgs)
2809+
protected FastTreePredictionWrapper(IHostEnvironment env, string name, Ensemble trainedEnsemble, int numFeatures, string innerArgs)
28102810
: base(env, name)
28112811
{
28122812
Host.CheckValue(trainedEnsemble, nameof(trainedEnsemble));

0 commit comments

Comments
 (0)