Add auto-sklearn to AutoML

minjk-bl · minjk-bl · commit f058b2992d07 · 2022-03-14T16:38:20.000+09:00
diff --git a/html/m_ml/model.html b/html/m_ml/model.html
@@ -19,12 +19,6 @@
     
                 </div>
             </div>
-            <div class="vp-grid-border-box vp-grid-col-110">
-                <label for="featureData" class="vp-orange-text">Feature data</label>
-                <input type="text" class="vp-input vp-state" id="featureData" placeholder="Select feature data" value="X_train"/>
-                <label for="targetData" class="vp-orange-text">Target data</label>
-                <input type="text" class="vp-input vp-state" id="targetData" placeholder="Select target data" value="y_train"/>
-            </div>
             <div class="vp-grid-border-box vp-grid-col-110">
                 <label for="allocateTo" class="vp-orange-text">Allocate to</label>
                 <input type="text" class="vp-input vp-state" id="allocateTo" placeholder="New variable name" value="model"/>
@@ -42,6 +36,15 @@
                     <!-- instanceEditor -->
                 </div>
             </div>
+            <div class="vp-grid-col-110 vp-model-instance-option-box">
+    
+            </div>
+            <!-- <div class="vp-grid-border-box vp-grid-col-110">
+                <label for="featureData" class="vp-orange-text">Feature data</label>
+                <input type="text" class="vp-input vp-state" id="featureData" placeholder="Select feature data" value="X_train"/>
+                <label for="targetData" class="vp-orange-text">Target data</label>
+                <input type="text" class="vp-input vp-state" id="targetData" placeholder="Select target data" value="y_train"/>
+            </div> -->
         </div>
     </div>
 </body>
diff --git a/js/com/com_Config.js b/js/com/com_Config.js
@@ -483,13 +483,13 @@ define([
      */
     Config.ML_DATA_DICT = {
         'Regression': [
-            'LinearRegression', 'SVR', 'DecisionTreeRegressor', 'RandomForestRegression', 'GradientBoostingRegressor', 'XGBRegressor', 'LGBMRegressor', 'CatBoostRegressor',
+            'LinearRegression', 'SVR', 'DecisionTreeRegressor', 'RandomForestRegressor', 'GradientBoostingRegressor', 'XGBRegressor', 'LGBMRegressor', 'CatBoostRegressor',
         ],
         'Classification': [
             'LogisticRegression', 'SVC', 'DecisionTreeClassifier', 'RandomForestClassifier', 'GradientBoostingClassifier', 'XGBClassifier', 'LGBMClassifier', 'CatBoostClassifier',
         ],
         'Auto ML': [
-            'TPOTRegression', 'TPOTClassifier'
+            'AutoSklearnRegressor', 'AutoSklearnClassifier', 'TPOTRegression', 'TPOTClassifier'
         ],
         'Clustering': [
             'KMeans', 'AgglomerativeClustering', 'GaussianMixture', 'DBSCAN',
