Updtae

Yu-Group · Mar 12, 2024 · b2d1a61 · b2d1a61
1 parent 53d8a02
commit b2d1a61
Show file tree

Hide file tree

Showing 6 changed files with 1,427 additions and 1,223 deletions.
diff --git a/feature_importance/01_run_ablation_classification.py b/feature_importance/01_run_ablation_classification.py
@@ -116,7 +116,9 @@ def compare_estimators(estimators: List[ModelConfig],
                     'model': model.name,
                     'fi': fi_est.name,
                     'splitting_strategy': splitting_strategy,
-                    'train_size': X_train.shape[0]
+                    'train_size': X_train.shape[0],
+                    'num_features': X_train.shape[1],
+                    'ablation_seed': seed
                 }
                 start = time.time()
                 if fi_est.name in ["LFI_with_raw", "LFI_without_raw"]:

diff --git a/feature_importance/01_run_ablation_regression.py b/feature_importance/01_run_ablation_regression.py
@@ -116,7 +116,9 @@ def compare_estimators(estimators: List[ModelConfig],
                     'model': model.name,
                     'fi': fi_est.name,
                     'splitting_strategy': splitting_strategy,
-                    'train_size': X_train.shape[0]
+                    'train_size': X_train.shape[0],
+                    'num_features': X_train.shape[1],
+                    'ablation_seed': seed
                 }
                 start = time.time()
                 if fi_est.name in ["LFI_with_raw", "LFI_without_raw"]: