Skip to content

Commit 9eab049

Browse files
srsaggamDmitry-A
authored andcommitted
change the type to float from double (dotnet#277)
1 parent 9542a66 commit 9eab049

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

src/Microsoft.ML.Auto/TrainerExtensions/SweepableParams.cs

+10-7
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ private static IEnumerable<SweepableParam> BuildAveragedLinearArgsParams()
1313
{
1414
return new SweepableParam[]
1515
{
16-
new SweepableDiscreteParam("LearningRate", new object[] { 0.01, 0.1, 0.5, 1.0 }),
16+
new SweepableDiscreteParam("LearningRate", new object[] { 0.01f, 0.1f, 0.5f, 1.0f}),
1717
new SweepableDiscreteParam("DecreaseLearningRate", new object[] { false, true }),
1818
new SweepableFloatParam("L2RegularizerWeight", 0.0f, 0.4f),
1919
};
@@ -31,12 +31,12 @@ private static IEnumerable<SweepableParam> BuildOnlineLinearArgsParams()
3131

3232
private static IEnumerable<SweepableParam> BuildTreeArgsParams()
3333
{
34-
return new SweepableParam[]
35-
{
34+
return new SweepableParam[]
35+
{
3636
new SweepableLongParam("NumLeaves", 2, 128, isLogScale: true, stepSize: 4),
3737
new SweepableDiscreteParam("MinDocumentsInLeafs", new object[] { 1, 10, 50 }),
3838
new SweepableDiscreteParam("NumTrees", new object[] { 20, 100, 500 }),
39-
};
39+
};
4040
}
4141

4242
private static IEnumerable<SweepableParam> BuildBoostedTreeArgsParams()
@@ -139,13 +139,15 @@ public static IEnumerable<SweepableParam> BuildSdcaParams()
139139
};
140140
}
141141

142-
public static IEnumerable<SweepableParam> BuildOrdinaryLeastSquaresParams() {
142+
public static IEnumerable<SweepableParam> BuildOrdinaryLeastSquaresParams()
143+
{
143144
return new SweepableParam[] {
144145
new SweepableDiscreteParam("L2Weight", new object[] { 1e-6f, 0.1f, 1f })
145146
};
146147
}
147148

148-
public static IEnumerable<SweepableParam> BuildSgdParams() {
149+
public static IEnumerable<SweepableParam> BuildSgdParams()
150+
{
149151
return new SweepableParam[] {
150152
new SweepableDiscreteParam("L2Weight", new object[] { 1e-7f, 5e-7f, 1e-6f, 5e-6f, 1e-5f }),
151153
new SweepableDiscreteParam("ConvergenceTolerance", new object[] { 1e-2f, 1e-3f, 1e-4f, 1e-5f }),
@@ -154,7 +156,8 @@ public static IEnumerable<SweepableParam> BuildSgdParams() {
154156
};
155157
}
156158

157-
public static IEnumerable<SweepableParam> BuildSymSgdParams() {
159+
public static IEnumerable<SweepableParam> BuildSymSgdParams()
160+
{
158161
return new SweepableParam[] {
159162
new SweepableDiscreteParam("NumberOfIterations", new object[] { 1, 5, 10, 20, 30, 40, 50 }),
160163
new SweepableDiscreteParam("LearningRate", new object[] { "<Auto>", 1e1f, 1e0f, 1e-1f, 1e-2f, 1e-3f }),

0 commit comments

Comments
 (0)