伪标签函数增加参数p.

4paradigm · Mar 14, 2022 · 540a992 · 540a992
1 parent 9fce06e
commit 540a992
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/autox/autox_competition/process_data/get_pseudo_label.py b/autox/autox_competition/process_data/get_pseudo_label.py
@@ -2,7 +2,8 @@
 from sklearn.model_selection import StratifiedKFold
 from tqdm import tqdm
 
-def get_pseudo_label(train, test, id_, target, used_cols):
+def get_pseudo_label(train, test, id_, target, used_cols, p = 0.99):
+    assert 0.5 < p < 1
     sub = test[[id_]]
     sub[target] = 0
 
@@ -15,7 +16,7 @@ def get_pseudo_label(train, test, id_, target, used_cols):
         pred = clf.predict_proba(test[used_cols])[:,1]
         sub[target] = sub[target] + pred / skf.n_splits
 
-    pseudo_test = sub[(sub[target] <= 0.01) | (sub[target] >= 0.99)].copy()
+    pseudo_test = sub[(sub[target] <= (1-p)) | (sub[target] >= p)].copy()
     pseudo_test.loc[pseudo_test[target] >= 0.5, target] = 1
     pseudo_test.loc[pseudo_test[target] < 0.5, target] = 0
     pseudo_test.index = range(len(pseudo_test))