Add SMOTE model info

minjk-bl · minjk-bl · commit ffec38cd1471 · 2023-09-13T17:39:06.000+09:00
diff --git a/visualpython/html/m_ml/modelInfo.html b/visualpython/html/m_ml/modelInfo.html
@@ -38,7 +38,7 @@
         </div>
         <div class="vp-grid-border-box">
             <div class="vp-multilang vp-ins-select-title">Options</div>
-            <div class="vp-ins-parameter-box vp-grid-col-95">
+            <div class="vp-ins-parameter-box vp-grid-col-130">
 
             </div>
         </div>
diff --git a/visualpython/js/m_ml/ModelInfo.js b/visualpython/js/m_ml/ModelInfo.js
@@ -514,6 +514,19 @@ define([
                             }
                         }
                     }
+                    if (modelType === 'SMOTE') {
+                        infos = {
+                            'get_feature_names_out': {
+                                name: 'get_feature_names_out',
+                                label: 'Get feature names',
+                                code: '${feature_names_allocate} = ${model}.get_feature_names_out()',
+                                description: 'Get output feature names for transformation.',
+                                options: [
+                                    { name: 'feature_names_allocate', label: 'Allocate to', component: ['input'], placeholder: 'New variable', value: 'features' }
+                                ]
+                            }
+                        }
+                    }
                     infos = {
                         ...infos,
                         'get_params': defaultInfos['get_params']
