Skip to content

Commit f59f671

Browse files
srsaggamDmitry-A
authored andcommitted
include using system in prediction class (dotnet#307)
* added using * fix test
1 parent a388a59 commit f59f671

File tree

3 files changed

+12
-9
lines changed

3 files changed

+12
-9
lines changed

src/mlnet.Test/ApprovalTests/ConsoleCodeGeneratorTests.PredictionCSFileContentTest.approved.txt

+1
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
//* *
55
//*****************************************************************************************
66

7+
using System;
78
using Microsoft.ML.Data;
89

910
namespace TestNamespace.Model.DataModels

src/mlnet/Templates/Console/PredictionClass.cs

+10-9
Original file line numberDiff line numberDiff line change
@@ -34,18 +34,19 @@ public virtual string TransformText()
3434
//* *
3535
//*****************************************************************************************
3636
37+
using System;
3738
using Microsoft.ML.Data;
3839
3940
namespace ");
4041

41-
#line 14 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
42+
#line 15 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
4243
this.Write(this.ToStringHelper.ToStringWithCulture(Namespace));
4344

4445
#line default
4546
#line hidden
4647
this.Write(".Model.DataModels\r\n{\r\n public class SamplePrediction\r\n {\r\n");
4748

48-
#line 18 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
49+
#line 19 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
4950
if("BinaryClassification".Equals(TaskType)){
5051

5152
#line default
@@ -54,7 +55,7 @@ namespace ");
5455
"/ its default value, which is the name of the field.\r\n [ColumnName(\"Predi" +
5556
"ctedLabel\")]\r\n public bool Prediction { get; set; }\r\n\r\n");
5657

57-
#line 24 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
58+
#line 25 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
5859
} if("MulticlassClassification".Equals(TaskType)){
5960

6061
#line default
@@ -63,34 +64,34 @@ namespace ");
6364
"/ its default value, which is the name of the field.\r\n [ColumnName(\"Predi" +
6465
"ctedLabel\")]\r\n public ");
6566

66-
#line 28 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
67+
#line 29 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
6768
this.Write(this.ToStringHelper.ToStringWithCulture(PredictionLabelType));
6869

6970
#line default
7071
#line hidden
7172
this.Write(" Prediction { get; set; }\r\n");
7273

73-
#line 29 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
74+
#line 30 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
7475
}
7576

7677
#line default
7778
#line hidden
7879

79-
#line 30 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
80+
#line 31 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
8081
if("MulticlassClassification".Equals(TaskType)){
8182

8283
#line default
8384
#line hidden
8485
this.Write(" public float[] Scores { get; set; }\r\n");
8586

86-
#line 32 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
87+
#line 33 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
8788
}else{
8889

8990
#line default
9091
#line hidden
9192
this.Write(" public float Score { get; set; }\r\n");
9293

93-
#line 34 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
94+
#line 35 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
9495
}
9596

9697
#line default
@@ -99,7 +100,7 @@ namespace ");
99100
return this.GenerationEnvironment.ToString();
100101
}
101102

102-
#line 37 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
103+
#line 38 "E:\src\machinelearning-automl\src\mlnet\Templates\Console\PredictionClass.tt"
103104

104105
public string TaskType {get;set;}
105106
public string PredictionLabelType {get;set;}

src/mlnet/Templates/Console/PredictionClass.tt

+1
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
//* *
1010
//*****************************************************************************************
1111

12+
using System;
1213
using Microsoft.ML.Data;
1314

1415
namespace <#= Namespace #>.Model.DataModels

0 commit comments

Comments
 (0)