Check and edit options on ModelInfo

minjk-bl · minjk-bl · commit c928f643e174 · 2022-06-15T17:19:25.000+09:00
diff --git a/js/m_ml/ModelInfo.js b/js/m_ml/ModelInfo.js
@@ -537,8 +537,6 @@ define([
                             ]
                         },
                         'permutation_importance': defaultInfos['permutation_importance'],
-                        'feature_importances': defaultInfos['feature_importances'],
-                        'plot_feature_importances': defaultInfos['plot_feature_importances'],
                         'Coefficient': {
                             name: 'coef_',
                             label: 'Coefficient',
@@ -558,6 +556,19 @@ define([
                             ]
                         }
                     }
+                    let svcList = [
+                        'DecisionTreeRegressor', 
+                        'RandomForestRegressor',
+                        'GradientBoostingRegressor', 
+                        'XGBRegressor', 'LGBMRegressor', 'CatBoostRegressor'
+                    ];
+                    if (svcList.includes(modelType)) {
+                        infos = {
+                            ...infos,
+                            'feature_importances': defaultInfos['feature_importances'],
+                            'plot_feature_importances': defaultInfos['plot_feature_importances']
+                        }
+                    }
                     break;
                 case 'Classification':
                     infos = {
@@ -615,7 +626,15 @@ define([
                     }
 
                     // feature importances
-                    if (modelType != 'LogisticRegression' && modelType != 'SVC') {
+                    let clfList = [
+                        'DecisionTreeClassifier', 
+                        'RandomForestClassifier', 
+                        'GradientBoostingClassifier', 
+                        'XGBClassifier', 
+                        'LGBMClassifier', 
+                        'CatBoostClassifier',
+                    ]
+                    if (clfList.includes(modelType)) {
                         infos = {
                             ...infos,
                             'feature_importances': defaultInfos['feature_importances'],
