Initial commit

2020-04-13 15:16:46 +01:00
commit 17a40cb95f
81 changed files with 10120 additions and 0 deletions
@@ -0,0 +1,16 @@
+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));
+        }
+    }
+}