Skip to content

Commit f8dd6f8

Browse files
authored
make pipeline object model & other public classes internal (dotnet#43)
1 parent ff599fd commit f8dd6f8

File tree

6 files changed

+18
-17
lines changed

6 files changed

+18
-17
lines changed

src/AutoML/API/MLContextAutoFitExtensions.cs

+6-6
Original file line numberDiff line numberDiff line change
@@ -184,9 +184,9 @@ public class BinaryClassificationItertionResult
184184
public readonly BinaryClassificationMetrics Metrics;
185185
public readonly ITransformer Model;
186186
public readonly IDataView ScoredValidationData;
187-
public readonly Pipeline Pipeline;
187+
internal readonly Pipeline Pipeline;
188188

189-
public BinaryClassificationItertionResult(ITransformer model, BinaryClassificationMetrics metrics, IDataView scoredValidationData, Pipeline pipeline)
189+
internal BinaryClassificationItertionResult(ITransformer model, BinaryClassificationMetrics metrics, IDataView scoredValidationData, Pipeline pipeline)
190190
{
191191
Model = model;
192192
ScoredValidationData = scoredValidationData;
@@ -200,9 +200,9 @@ public class MulticlassClassificationIterationResult
200200
public readonly MultiClassClassifierMetrics Metrics;
201201
public readonly ITransformer Model;
202202
public readonly IDataView ScoredValidationData;
203-
public readonly Pipeline Pipeline;
203+
internal readonly Pipeline Pipeline;
204204

205-
public MulticlassClassificationIterationResult(ITransformer model, MultiClassClassifierMetrics metrics, IDataView scoredValidationData, Pipeline pipeline)
205+
internal MulticlassClassificationIterationResult(ITransformer model, MultiClassClassifierMetrics metrics, IDataView scoredValidationData, Pipeline pipeline)
206206
{
207207
Model = model;
208208
Metrics = metrics;
@@ -216,9 +216,9 @@ public class RegressionIterationResult
216216
public readonly RegressionMetrics Metrics;
217217
public readonly ITransformer Model;
218218
public readonly IDataView ScoredValidationData;
219-
public readonly Pipeline Pipeline;
219+
internal readonly Pipeline Pipeline;
220220

221-
public RegressionIterationResult(ITransformer model, RegressionMetrics metrics, IDataView scoredValidationData, Pipeline pipeline)
221+
internal RegressionIterationResult(ITransformer model, RegressionMetrics metrics, IDataView scoredValidationData, Pipeline pipeline)
222222
{
223223
Model = model;
224224
Metrics = metrics;

src/AutoML/API/Pipeline.cs

+8-7
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
namespace Microsoft.ML.Auto
99
{
10-
public class Pipeline
10+
internal class Pipeline
1111
{
1212
public PipelineNode[] Nodes { get; set; }
1313

@@ -28,7 +28,7 @@ public IEstimator<ITransformer> ToEstimator()
2828
}
2929
}
3030

31-
public class PipelineNode
31+
internal class PipelineNode
3232
{
3333
public string Name { get; set; }
3434
public PipelineNodeType NodeType { get; set; }
@@ -65,13 +65,13 @@ internal PipelineNode()
6565
}
6666
}
6767

68-
public enum PipelineNodeType
68+
internal enum PipelineNodeType
6969
{
7070
Transform,
7171
Trainer
7272
}
7373

74-
public class CustomProperty
74+
internal class CustomProperty
7575
{
7676
public string Name { get; set; }
7777
public IDictionary<string, object> Properties { get; set; }
@@ -87,9 +87,8 @@ internal CustomProperty()
8787
}
8888
}
8989

90-
public class PipelineRunResult
90+
internal class PipelineRunResult
9191
{
92-
public readonly Pipeline Pipeline;
9392
public readonly double Score;
9493

9594
/// <summary>
@@ -98,7 +97,9 @@ public class PipelineRunResult
9897
/// </summary>
9998
public readonly bool RunSucceded;
10099

101-
public PipelineRunResult(Pipeline pipeline, double score, bool runSucceeded)
100+
internal readonly Pipeline Pipeline;
101+
102+
internal PipelineRunResult(Pipeline pipeline, double score, bool runSucceeded)
102103
{
103104
Pipeline = pipeline;
104105
Score = score;

src/AutoML/AutoFitter/OptimizingMetric.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
namespace Microsoft.ML.Auto
88
{
9-
public enum OptimizingMetric
9+
internal enum OptimizingMetric
1010
{
1111
Auc,
1212
Accuracy,

src/AutoML/EstimatorExtensions/EstimatorExtensionCatalog.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
namespace Microsoft.ML.Auto
99
{
10-
public enum EstimatorName
10+
internal enum EstimatorName
1111
{
1212
ColumnConcatenating,
1313
ColumnCopying,

src/AutoML/TaskKind.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
namespace Microsoft.ML.Auto
66
{
7-
public enum TaskKind
7+
internal enum TaskKind
88
{
99
BinaryClassification,
1010
MulticlassClassification,

src/AutoML/TrainerExtensions/TrainerExtensionUtil.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
namespace Microsoft.ML.Auto
1212
{
13-
public enum TrainerName
13+
internal enum TrainerName
1414
{
1515
AveragedPerceptronBinary,
1616
AveragedPerceptronOva,

0 commit comments

Comments
 (0)