Skip to content

Commit 3f1e454

Browse files
author
Tom Finley
committed
Use normalize convenience constructors.
1 parent 7fb15ae commit 3f1e454

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
lines changed

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

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -523,10 +523,7 @@ public static bool AddNormalizerIfNeeded(IHostEnvironment env, IChannel ch, ITra
523523
}
524524
ch.Info("Automatically adding a MinMax normalization transform, use 'norm=Warn' or 'norm=No' to turn this behavior off.");
525525
IDataView ApplyNormalizer(IHostEnvironment innerEnv, IDataView input)
526-
=> NormalizeTransform.Create(innerEnv, new NormalizeTransform.MinMaxArguments()
527-
{
528-
Column = new[] { new NormalizeTransform.AffineColumn { Source = featureColumn, Name = featureColumn } }
529-
}, input);
526+
=> NormalizeTransform.CreateMinMaxNormalizer(innerEnv, input, featureColumn);
530527

531528
if (view is IDataLoader loader)
532529
view = CompositeDataLoader.ApplyTransform(env, loader, tag: null, creationArgs: null, ApplyNormalizer);

src/Microsoft.ML.Data/Transforms/NormalizeTransform.cs

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -222,10 +222,7 @@ public static bool CreateIfNeeded(IHostEnvironment env, ref RoleMappedData data,
222222
var featInfo = data.Schema.Feature;
223223
env.AssertValue(featInfo); // Should be defined, if FEaturesAreNormalized returned a definite value.
224224

225-
var view = Create(env, new MinMaxArguments()
226-
{
227-
Column = new[] { new AffineColumn() { Name = featInfo.Name, Source = featInfo.Name } }
228-
}, data.Data);
225+
var view = CreateMinMaxNormalizer(env, data.Data, name: featInfo.Name);
229226
data = RoleMappedData.Create(view, data.Schema.GetColumnRoleNames());
230227
return true;
231228
}

0 commit comments

Comments
 (0)