leavoigt commited on
Commit
aeb4fb8
1 Parent(s): f938459

Update appStore/target.py

Browse files
Files changed (1) hide show
  1. appStore/target.py +3 -0
appStore/target.py CHANGED
@@ -62,6 +62,9 @@ def app():
62
  # Load the existing dataset
63
  df = st.session_state.key1
64
 
 
 
 
65
  # Load the classifier model
66
  classifier = load_targetClassifier(classifier_name=params['model_name'])
67
 
 
62
  # Load the existing dataset
63
  df = st.session_state.key1
64
 
65
+ # Filter out all paragraphs that do not have a reference to groups
66
+ df = df[~df['Vulnerability Label'].apply(lambda labels: 'Other' in labels)]
67
+
68
  # Load the classifier model
69
  classifier = load_targetClassifier(classifier_name=params['model_name'])
70