diff --git a/src/Microsoft.Data.Analysis/DataFrameColumn.cs b/src/Microsoft.Data.Analysis/DataFrameColumn.cs index b88ffc93d7..81ee51b888 100644 --- a/src/Microsoft.Data.Analysis/DataFrameColumn.cs +++ b/src/Microsoft.Data.Analysis/DataFrameColumn.cs @@ -410,7 +410,7 @@ protected Dictionary> GetGroupedOccurrences(DataFr public virtual bool HasDescription() => false; /// - /// Returns a containing the DataType and Length of this column + /// Returns a containing the DataType and Length of this column. /// public virtual StringDataFrameColumn Info() { diff --git a/src/Microsoft.ML.AutoML/TrainerExtensions/BinaryTrainerExtensions.cs b/src/Microsoft.ML.AutoML/TrainerExtensions/BinaryTrainerExtensions.cs index 493a6dc731..315379dea3 100644 --- a/src/Microsoft.ML.AutoML/TrainerExtensions/BinaryTrainerExtensions.cs +++ b/src/Microsoft.ML.AutoML/TrainerExtensions/BinaryTrainerExtensions.cs @@ -226,6 +226,26 @@ public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable sweepParams, ColumnInformation columnInfo) + { + return TrainerExtensionUtil.BuildPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams, + columnInfo.LabelColumnName); + } + } + internal class FieldAwareFactorizationMachineBinaryExtension : ITrainerExtension + { + public IEnumerable GetHyperparamSweepRanges() + { + return SweepableParams.BuildSdcaParams(); + } + + public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable sweepParams, + ColumnInformation columnInfo, IDataView validationSet) + { + var options = TrainerExtensionUtil.CreateOptions(sweepParams, columnInfo.LabelColumnName); + return mlContext.BinaryClassification.Trainers.FieldAwareFactorizationMachine(options); + } + public PipelineNode CreatePipelineNode(IEnumerable sweepParams, ColumnInformation columnInfo) { return TrainerExtensionUtil.BuildPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams,