search

motefly / DeepGBM

SIGKDD'2019: DeepGBM: A Deep Learning Framework Distilled by GBDT for Online Prediction Tasks
647 stars 135 forks source link

The predicted y output for the nips_A dataset is not binary #29

Open nurmanmus opened 4 years ago

nurmanmus commented 4 years ago

I tried to replicate the experiment using nipsA_deepgbm_offline script but when i tried to print the output of the predictions (i.e., pred_y), the predicted values given are real number as follows:

pred_y: [0.34316275 0.36292914 0.3552964 0.35970888 0.35918367 0.3515028 0.36127064 0.34485954 0.37243855 0.3597494 0.37358335 0.35961217 0.3446463 0.3496978 0.3566822 0.35322982 0.36830378 0.37577894 0.3774116 0.361733 0.34249693 0.36554664 0.36565682 0.3529494 0.3586624 0.35545474 0.35036924 0.37629476 0.36555907 0.37944567 0.37318298 0.37047318 0.36648336 0.3657822 0.36535558 0.37917492 0.3766928 0.37294027 0.365851 0.36476082 0.3654274 0.37421528 0.35136348 0.3707069 0.37431964 0.38356563 0.35120273 0.37153876 0.3950685 0.3748232 0.36651853 0.35673445 0.36918446 0.36931384 0.36340126 0.3641296 0.38208184 0.3779632 0.36068133 0.34996226 0.34128729 0.3601104 0.35272548 0.34229857 0.35786942 0.352486 0.34367353 0.34292746 0.3950783 0.36609793 0.3616757 0.38065642

Isn't the predicted output suppose to be 0 or 1? Can you please advise.

jaronn commented 4 years ago

this is the propability of the label