/// <summary>
/// Public create method corresponding to SignatureDataTransform.
/// </summary>
public static NormalizeTransform Create(IHostEnvironment env, SupervisedBinArguments args, IDataView input)
{
Contracts.CheckValue(env, nameof(env));
var h = env.Register("Normalize(SupervisedBin)");
NormalizeTransform func;
using (var ch = h.Start("Training"))
{
var labelColumnId = SupervisedBinUtils.GetLabelColumnId(ch, input.Schema, args.LabelColumn);
func = Create(h, args, input, SupervisedBinUtils.CreateBuilder, labelColumnId);
ch.Done();
}
return(func);
}
public static IColumnFunctionBuilder CreateBuilder(SupervisedBinArguments args, IHost host,
int icol, int srcIndex, ColumnType srcType, IRowCursor cursor)
{
Contracts.AssertValue(host);
host.AssertValue(args);
// checking for label column
host.CheckUserArg(!string.IsNullOrWhiteSpace(args.LabelColumn), nameof(args.LabelColumn), "Must specify the label column name");
int labelColumnId = GetLabelColumnId(host, cursor.Schema, args.LabelColumn);
var labelColumnType = cursor.Schema.GetColumnType(labelColumnId);
if (labelColumnType.IsKey)
{
host.CheckUserArg(labelColumnType.KeyCount > 0, nameof(args.LabelColumn), "Label column must have a known cardinality");
}
else
{
host.CheckUserArg(labelColumnType.IsNumber, nameof(args.LabelColumn), "Label column must be a number or a key type");
}
if (srcType.IsNumber)
{
if (srcType == NumberType.R4)
{
return(Sng.SupervisedBinOneColumnFunctionBuilder.Create(args, host, icol, srcIndex, labelColumnId, cursor));
}
if (srcType == NumberType.R8)
{
return(Dbl.SupervisedBinOneColumnFunctionBuilder.Create(args, host, icol, srcIndex, labelColumnId, cursor));
}
}
if (srcType.IsVector && srcType.ItemType.IsNumber)
{
if (srcType.ItemType == NumberType.R4)
{
return(Sng.SupervisedBinVecColumnFunctionBuilder.Create(args, host, icol, srcIndex, labelColumnId, cursor));
}
if (srcType.ItemType == NumberType.R8)
{
return(Dbl.SupervisedBinVecColumnFunctionBuilder.Create(args, host, icol, srcIndex, labelColumnId, cursor));
}
}
throw host.ExceptUserArg(nameof(args.Column), "Wrong column type for column {0}. Expected: R4, R8, Vec<R4, n> or Vec<R8, n>. Got: {1}.", args.Column[icol].Source, srcType.ToString());
}
