Ganmin Yin

icon indicating a website link https://www.sunshineyin.site/ icon indicating an email [email protected]

icon company Peking University icon location Beijing, China icon location A fool on the way of learning.

Results 2 comments of Ganmin Yin

About the softmax layer of the model architecture

Any reply? Thanks.

About the softmax layer of the model architecture

Dear Luca, Yeah, I have found your code based on your reply. But I still have some questions. In your `predict_proba` function of `od_models.py`, https://github.com/scikit-mobility/DeepGravity/blob/8693536dac5046f7c484711b3665d73610d83e4b/deepgravity/models/od_models.py#L77 Your input to the softmax...