Skip to content

Commit 5f44e9b

Browse files
committed
Internalize IEvaluator implementations.
1 parent e08a329 commit 5f44e9b

11 files changed

+41
-21
lines changed

src/Microsoft.ML.Data/Evaluators/AnomalyDetectionEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,8 @@
2222

2323
namespace Microsoft.ML.Data
2424
{
25-
public sealed class AnomalyDetectionEvaluator : EvaluatorBase<AnomalyDetectionEvaluator.Aggregator>
25+
[BestFriend]
26+
internal sealed class AnomalyDetectionEvaluator : EvaluatorBase<AnomalyDetectionEvaluator.Aggregator>
2627
{
2728
public sealed class Arguments
2829
{
@@ -577,7 +578,8 @@ public void Finish()
577578
}
578579
}
579580

580-
public sealed class AnomalyDetectionMamlEvaluator : MamlEvaluatorBase
581+
[BestFriend]
582+
internal sealed class AnomalyDetectionMamlEvaluator : MamlEvaluatorBase
581583
{
582584
public sealed class Arguments : ArgumentsBase
583585
{

src/Microsoft.ML.Data/Evaluators/AucAggregator.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
namespace Microsoft.ML.Data
1212
{
13-
public abstract partial class EvaluatorBase<TAgg>
13+
internal abstract partial class EvaluatorBase<TAgg>
1414
{
1515
internal abstract class AucAggregatorBase
1616
{

src/Microsoft.ML.Data/Evaluators/BinaryClassifierEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@
2828

2929
namespace Microsoft.ML.Data
3030
{
31-
public sealed class BinaryClassifierEvaluator : RowToRowEvaluatorBase<BinaryClassifierEvaluator.Aggregator>
31+
[BestFriend]
32+
internal sealed class BinaryClassifierEvaluator : RowToRowEvaluatorBase<BinaryClassifierEvaluator.Aggregator>
3233
{
3334
public sealed class Arguments
3435
{
@@ -1116,7 +1117,8 @@ private void CheckInputColumnTypes(Schema schema)
11161117
}
11171118
}
11181119

1119-
public sealed class BinaryClassifierMamlEvaluator : MamlEvaluatorBase
1120+
[BestFriend]
1121+
internal sealed class BinaryClassifierMamlEvaluator : MamlEvaluatorBase
11201122
{
11211123
public class Arguments : ArgumentsBase
11221124
{

src/Microsoft.ML.Data/Evaluators/ClusteringEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ namespace Microsoft.ML.Data
2929
{
3030
using Conditional = System.Diagnostics.ConditionalAttribute;
3131

32-
public sealed class ClusteringEvaluator : RowToRowEvaluatorBase<ClusteringEvaluator.Aggregator>
32+
[BestFriend]
33+
internal sealed class ClusteringEvaluator : RowToRowEvaluatorBase<ClusteringEvaluator.Aggregator>
3334
{
3435
public sealed class Arguments
3536
{
@@ -758,7 +759,8 @@ private void CheckInputColumnTypes(Schema schema)
758759
}
759760
}
760761

761-
public sealed class ClusteringMamlEvaluator : MamlEvaluatorBase
762+
[BestFriend]
763+
internal sealed class ClusteringMamlEvaluator : MamlEvaluatorBase
762764
{
763765
public class Arguments : ArgumentsBase
764766
{

src/Microsoft.ML.Data/Evaluators/EvaluatorBase.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@ namespace Microsoft.ML.Data
1717
/// <see cref="GetPerInstanceMetricsCore"/>. Note that the input <see cref="RoleMappedData"/> is assumed to contain all the column
1818
/// roles needed for evaluation, including the score column.
1919
/// </summary>
20-
public abstract partial class EvaluatorBase<TAgg> : IEvaluator
20+
[BestFriend]
21+
internal abstract partial class EvaluatorBase<TAgg> : IEvaluator
2122
where TAgg : EvaluatorBase<TAgg>.AggregatorBase
2223
{
2324
protected readonly IHost Host;
@@ -446,7 +447,8 @@ public override IEnumerable<TAgg> GetAll()
446447
}
447448
}
448449

449-
public abstract class RowToRowEvaluatorBase<TAgg> : EvaluatorBase<TAgg>
450+
[BestFriend]
451+
internal abstract class RowToRowEvaluatorBase<TAgg> : EvaluatorBase<TAgg>
450452
where TAgg : EvaluatorBase<TAgg>.AggregatorBase
451453
{
452454
[BestFriend]

src/Microsoft.ML.Data/Evaluators/MultiClassClassifierEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@
2828

2929
namespace Microsoft.ML.Data
3030
{
31-
public sealed class MultiClassClassifierEvaluator : RowToRowEvaluatorBase<MultiClassClassifierEvaluator.Aggregator>
31+
[BestFriend]
32+
internal sealed class MultiClassClassifierEvaluator : RowToRowEvaluatorBase<MultiClassClassifierEvaluator.Aggregator>
3233
{
3334
public sealed class Arguments
3435
{
@@ -822,7 +823,8 @@ private void CheckInputColumnTypes(Schema schema)
822823
}
823824
}
824825

825-
public sealed class MultiClassMamlEvaluator : MamlEvaluatorBase
826+
[BestFriend]
827+
internal sealed class MultiClassMamlEvaluator : MamlEvaluatorBase
826828
{
827829
public class Arguments : ArgumentsBase
828830
{

src/Microsoft.ML.Data/Evaluators/MultiOutputRegressionEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@
2828

2929
namespace Microsoft.ML.Data
3030
{
31-
public sealed class MultiOutputRegressionEvaluator : RegressionLossEvaluatorBase<MultiOutputRegressionEvaluator.Aggregator>
31+
[BestFriend]
32+
internal sealed class MultiOutputRegressionEvaluator : RegressionLossEvaluatorBase<MultiOutputRegressionEvaluator.Aggregator>
3233
{
3334
public sealed class Arguments : ArgumentsBase
3435
{
@@ -609,7 +610,8 @@ private ValueGetter<VBuffer<ReadOnlyMemory<char>>> CreateSlotNamesGetter(Schema
609610
}
610611
}
611612

612-
public sealed class MultiOutputRegressionMamlEvaluator : MamlEvaluatorBase
613+
[BestFriend]
614+
internal sealed class MultiOutputRegressionMamlEvaluator : MamlEvaluatorBase
613615
{
614616
public sealed class Arguments : ArgumentsBase
615617
{

src/Microsoft.ML.Data/Evaluators/QuantileRegressionEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@
2525

2626
namespace Microsoft.ML.Data
2727
{
28-
public sealed class QuantileRegressionEvaluator :
28+
[BestFriend]
29+
internal sealed class QuantileRegressionEvaluator :
2930
RegressionEvaluatorBase<QuantileRegressionEvaluator.Aggregator, VBuffer<Float>, VBuffer<Double>>
3031
{
3132
public sealed class Arguments : ArgumentsBase
@@ -457,7 +458,8 @@ private void CheckInputColumnTypes(Schema schema)
457458
}
458459
}
459460

460-
public sealed class QuantileRegressionMamlEvaluator : MamlEvaluatorBase
461+
[BestFriend]
462+
internal sealed class QuantileRegressionMamlEvaluator : MamlEvaluatorBase
461463
{
462464
public sealed class Arguments : ArgumentsBase
463465
{

src/Microsoft.ML.Data/Evaluators/RankerEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,8 @@
2727

2828
namespace Microsoft.ML.Data
2929
{
30-
public sealed class RankerEvaluator : EvaluatorBase<RankerEvaluator.Aggregator>
30+
[BestFriend]
31+
internal sealed class RankerEvaluator : EvaluatorBase<RankerEvaluator.Aggregator>
3132
{
3233
public sealed class Arguments
3334
{
@@ -836,7 +837,8 @@ public RowCursorState(int truncationLevel)
836837
}
837838
}
838839

839-
public sealed class RankerMamlEvaluator : MamlEvaluatorBase
840+
[BestFriend]
841+
internal sealed class RankerMamlEvaluator : MamlEvaluatorBase
840842
{
841843
public sealed class Arguments : ArgumentsBase
842844
{

src/Microsoft.ML.Data/Evaluators/RegressionEvaluator.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@
2424

2525
namespace Microsoft.ML.Data
2626
{
27-
public sealed class RegressionEvaluator :
27+
[BestFriend]
28+
internal sealed class RegressionEvaluator :
2829
RegressionEvaluatorBase<RegressionEvaluator.Aggregator, Float, Double>
2930
{
3031
public sealed class Arguments : ArgumentsBase
@@ -327,7 +328,8 @@ private void CheckInputColumnTypes(Schema schema)
327328
}
328329
}
329330

330-
public sealed class RegressionMamlEvaluator : MamlEvaluatorBase
331+
[BestFriend]
332+
internal sealed class RegressionMamlEvaluator : MamlEvaluatorBase
331333
{
332334
public sealed class Arguments : ArgumentsBase
333335
{

src/Microsoft.ML.Data/Evaluators/RegressionEvaluatorBase.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@
1010

1111
namespace Microsoft.ML.Data
1212
{
13-
public abstract class RegressionLossEvaluatorBase<TAgg> : RowToRowEvaluatorBase<TAgg>
13+
[BestFriend]
14+
internal abstract class RegressionLossEvaluatorBase<TAgg> : RowToRowEvaluatorBase<TAgg>
1415
where TAgg : EvaluatorBase<TAgg>.AggregatorBase
1516
{
1617
public abstract class ArgumentsBase
@@ -35,7 +36,8 @@ protected RegressionLossEvaluatorBase(ArgumentsBase args, IHostEnvironment env,
3536
}
3637
}
3738

38-
public abstract class RegressionEvaluatorBase<TAgg, TScore, TMetrics> : RegressionLossEvaluatorBase<TAgg>
39+
[BestFriend]
40+
internal abstract class RegressionEvaluatorBase<TAgg, TScore, TMetrics> : RegressionLossEvaluatorBase<TAgg>
3941
where TAgg : RegressionEvaluatorBase<TAgg, TScore, TMetrics>.RegressionAggregatorBase
4042
{
4143
[BestFriend]

0 commit comments

Comments
 (0)