clean

Author: zewditu

src/Microsoft.ML.AutoML/CodeGen/question-answering.json renamed to src/Microsoft.ML.AutoML/CodeGen/question_answering_search_space.json

@@ -1,14 +1,14 @@
 {
   "$schema": "./search-space-schema.json#",
-  "name": "question_answering _option",
+  "name": "question_answering_option",
   "search_space": [
     {
       "name": "ContextColumnName",
       "type": "string",
       "default": "Context"
     },
     {
-      "name": " QuestionColumnName",
+      "name": "QuestionColumnName",
       "type": "string",
       "default": "Question"
     },
@@ -42,6 +42,11 @@
       "type": "integer",
       "default": 10
     },
+    {
+      "name": "TopKAnswers",
+      "type": "integer",
+      "default": 3
+    },
     {
       "name": "Architecture",
       "type": "bertArchitecture",

src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json

@@ -147,7 +147,7 @@
         "text_classification_option",
         "sentence_similarity_option",
         "object_detection_option",
-        "question_answering _option"
+        "question_answering_option"
       ]
     },
     "option_name": {
@@ -211,7 +211,13 @@
         "Steps",
         "MaxEpoch",
         "InitLearningRate",
-        "WeightDecay"
+        "WeightDecay",
+        "ContextColumnName",
+        "QuestionColumnName",
+        "TrainingAnswerColumnName",
+        "AnswerIndexStartColumnName",
+        "predictedAnswerColumnName",
+        "TopKAnswers"
       ]
     },
     "option_type": {

src/Microsoft.ML.AutoML/Microsoft.ML.AutoML.csproj

@@ -69,7 +69,7 @@
     <AdditionalFiles Include="CodeGen\code_gen_flag.json" />
     <AdditionalFiles Include="CodeGen\*-estimators.json" />
   </ItemGroup>
-
+  
   <ItemGroup>
     <EmbeddedResource Include="Tuner\Portfolios.json">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>

src/Microsoft.ML.AutoML/CodeGen/QuestionAnswering.cs renamed to src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/QuestionAnswering.cs

@@ -7,9 +7,9 @@
 
 namespace Microsoft.ML.AutoML.CodeGen
 {
-    internal class QuestionAnsweringMulti
+    internal partial class QuestionAnsweringMulti
     {
-        public IEstimator<ITransformer> BuildFromOption(MLContext context, QATrainer.Options param)
+        public override IEstimator<ITransformer> BuildFromOption(MLContext context, QuestionAnsweringOption param)
         {
             return context.MulticlassClassification.Trainers.QuestionAnswer(
                    contextColumnName: param.ContextColumnName,
@@ -19,9 +19,10 @@ public IEstimator<ITransformer> BuildFromOption(MLContext context, QATrainer.Opt
                    predictedAnswerColumnName: param.PredictedAnswerColumnName,
                    scoreColumnName: param.ScoreColumnName,
                    batchSize: param.BatchSize,
-                   maxEpochs: param.MaxEpoch,
+                   maxEpochs: param.MaxEpochs,
                    topK: param.TopKAnswers,
                    architecture: BertArchitecture.Roberta);
         }
+
     }
 }