17 lines
615 B
C#
17 lines
615 B
C#
using Microsoft.ML;
|
|
using Microsoft.ML.Data;
|
|
using Microsoft.ML.Transforms;
|
|
|
|
namespace AutoBet.Learn.Extensions
|
|
{
|
|
public static class EstimatorExtensions
|
|
{
|
|
//Converts string to one hot encoding vectors
|
|
public static EstimatorChain<OneHotEncodingTransformer> EncodeColumn(this IEstimator<ITransformer> pipeline, MLContext mlContext, string column)
|
|
{
|
|
var catergoricalTransforms = mlContext.Transforms.Categorical;
|
|
return pipeline.Append(catergoricalTransforms.OneHotEncoding(outputColumnName: column + "Encoded", inputColumnName: column));
|
|
}
|
|
}
|
|
}
|