search

Gurobi / gurobi-machinelearning

Formulate trained predictors in Gurobi models
https://gurobi-machinelearning.readthedocs.io/
Apache License 2.0
215 stars 44 forks source link

New models for softmax #355

Open pobonomo opened 1 month ago

pobonomo commented 1 month ago

Do new models for softmax

mmaaz-git commented 2 weeks ago

Hi Pierre: just wanted to check if you have an estimate for when you think this will be merged. I'd love to use it! No pressure at all. Or as you indicate it's safe to test, so I could locally clone this PR and play around with it? Thanks.

pobonomo commented 2 weeks ago

Hi Pierre: just wanted to check if you have an estimate for when you think this will be merged. I'd love to use it! No pressure at all. Or as you indicate it's safe to test, so I could locally clone this PR and play around with it? Thanks.

Hi,

I have no estimate currently. The models are working I think but it's a big change and there is a lot of cleanup.

Note that what is implemented so far is using softmax for scikit-learn only: multinomial logistic regression and neural networks. I didn't implement counter-parts for Keras and Pytorch yet (the code should be very simple but there would be a lot of testing).

I think that it is usable though so if you want to play with it you could try (and more testing cannot hurt).

Best, Pierre

mmaaz-git commented 2 weeks ago

Thanks, will do!