Skip to content

Commit 0f29ff1

Browse files
committed
Fixing importance script
1 parent 1388aba commit 0f29ff1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

bin/filter_feature_importance.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,9 @@ def format_delta(value: float) -> str:
172172
def main() -> None:
173173
args = parse_args()
174174
data = pd.read_csv(args.data, sep="\t")
175+
# Ensure any NaN labels are treated as unlabeled ("None")
176+
# This prevents NaN values from passing through the != "None" filter in semiSupRandomForest
177+
data["label"] = data["label"].fillna("None")
175178

176179
feature_list = resolve_feature_list(data, args.features, args.ignore)
177180
if not feature_list:

0 commit comments

Comments
 (0)