Skip to content

Instantly share code, notes, and snippets.

@a-agmon

a-agmon/pu_est2.py

Last active March 2, 2020 12:36
Show Gist options
  • Save a-agmon/da38e99d9ee3c8787979b25a69d828e5 to your computer and use it in GitHub Desktop.
Save a-agmon/da38e99d9ee3c8787979b25a69d828e5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pu_est2.py
pu_estimator, probs1y1 = fit_PU_estimator(
x_train,
y_train,
0.2,
xgb.XGBClassifier())
predicted_s = pu_estimator.predict_proba(x_train)
predicted_s = predicted_s[:,1]
predicted_y = predicted_s / probs1y1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment