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));
        }
    }
}