forked from dotnet/machinelearning
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathNAHandling.cs
More file actions
99 lines (92 loc) · 5.5 KB
/
Copy pathNAHandling.cs
File metadata and controls
99 lines (92 loc) · 5.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using Microsoft.ML.EntryPoints;
using Microsoft.ML.Transforms;
[assembly: EntryPointModule(typeof(NAHandling))]
namespace Microsoft.ML.Transforms
{
public static class NAHandling
{
[TlcModule.EntryPoint(Name = "Transforms.MissingValuesDropper",
Desc = MissingValueDroppingTransformer.Summary,
UserName = MissingValueDroppingTransformer.FriendlyName,
ShortName = MissingValueDroppingTransformer.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/doc.xml' path='doc/members/member[@name=""NADrop""]/*' />",
@"<include file='../Microsoft.ML.Transforms/doc.xml' path='doc/members/example[@name=""NADrop""]/*' />" })]
public static CommonOutputs.TransformOutput Drop(IHostEnvironment env, MissingValueDroppingTransformer.Arguments input)
{
var h = EntryPointUtils.CheckArgsAndCreateHost(env, MissingValueDroppingTransformer.ShortName, input);
var xf = MissingValueDroppingTransformer.Create(h, input, input.Data);
return new CommonOutputs.TransformOutput()
{
Model = new TransformModelImpl(h, xf, input.Data),
OutputData = xf
};
}
[TlcModule.EntryPoint(Name = "Transforms.MissingValuesRowDropper",
Desc = NAFilter.Summary,
UserName = NAFilter.FriendlyName,
ShortName = NAFilter.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Data/Transforms/doc.xml' path='doc/members/member[@name=""NAFilter""]/*' />",
@"<include file='../Microsoft.ML.Data/Transforms/doc.xml' path='doc/members/example[@name=""NAFilter""]/*' />"})]
public static CommonOutputs.TransformOutput Filter(IHostEnvironment env, NAFilter.Arguments input)
{
var h = EntryPointUtils.CheckArgsAndCreateHost(env, NAFilter.ShortName, input);
var xf = new NAFilter(h, input, input.Data);
return new CommonOutputs.TransformOutput()
{
Model = new TransformModelImpl(h, xf, input.Data),
OutputData = xf
};
}
[TlcModule.EntryPoint(Name = "Transforms.MissingValueHandler",
Desc = MissingValueHandlingTransformer.Summary,
UserName = MissingValueHandlingTransformer.FriendlyName,
ShortName = MissingValueHandlingTransformer.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Data/Transforms/doc.xml' path='doc/members/member[@name=""NAHandle""]/*' />",
@"<include file='../Microsoft.ML.Data/Transforms/doc.xml' path='doc/members/example[@name=""NAHandle""]/*' />" })]
public static CommonOutputs.TransformOutput Handle(IHostEnvironment env, MissingValueHandlingTransformer.Arguments input)
{
var h = EntryPointUtils.CheckArgsAndCreateHost(env, "NAHandle", input);
var xf = MissingValueHandlingTransformer.Create(h, input, input.Data);
return new CommonOutputs.TransformOutput()
{
Model = new TransformModelImpl(h, xf, input.Data),
OutputData = xf
};
}
[TlcModule.EntryPoint(Name = "Transforms.MissingValueIndicator",
Desc = MissingValueIndicatorTransformer.Summary,
UserName = MissingValueIndicatorTransformer.FriendlyName,
ShortName = MissingValueIndicatorTransformer.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/doc.xml' path='doc/members/member[@name=""NAIndicator""]/*' />",
@"<include file='../Microsoft.ML.Transforms/doc.xml' path='doc/members/example[@name=""NAIndicator""]/*' />"})]
public static CommonOutputs.TransformOutput Indicator(IHostEnvironment env, MissingValueIndicatorTransformer.Arguments input)
{
var h = EntryPointUtils.CheckArgsAndCreateHost(env, "NAIndicator", input);
var xf = new MissingValueIndicatorTransformer(h, input).Transform(input.Data);
return new CommonOutputs.TransformOutput()
{
Model = new TransformModelImpl(h, xf, input.Data),
OutputData = xf
};
}
[TlcModule.EntryPoint(Name = "Transforms.MissingValueSubstitutor",
Desc = MissingValueReplacingTransformer.Summary,
UserName = MissingValueReplacingTransformer.FriendlyName,
ShortName = MissingValueReplacingTransformer.ShortName,
XmlInclude = new[] { @"<include file='../Microsoft.ML.Transforms/doc.xml' path='doc/members/member[@name=""NAReplace""]/*' />",
@"<include file='../Microsoft.ML.Transforms/doc.xml' path='doc/members/example[@name=""NAReplace""]/*' />"})]
public static CommonOutputs.TransformOutput Replace(IHostEnvironment env, MissingValueReplacingTransformer.Arguments input)
{
var h = EntryPointUtils.CheckArgsAndCreateHost(env, "NAReplace", input);
var xf = MissingValueReplacingTransformer.Create(h, input, input.Data);
return new CommonOutputs.TransformOutput()
{
Model = new TransformModelImpl(h, xf, input.Data),
OutputData = xf
};
}
}
}