k9k2/qsgd

SGD and Ordered SGD codes for deep learning, SVM, and logistic regression

qsgd's Issues

How to draw the decision boundary for Figure 1 (a) and (b)?

Hi Kenji,

I would like to draw a decision boundary for figure 1 (a) and (b) use your Linear Regression model with 2 dimensions output.

However, there is no relevant code in your plot.py. Could you please let me know how to draw the decision boundary?

implementation in huggingface trainer

Could you give an example of how this can be implemented in the huggingface trainer class? I'm not sure how q can be added and allow to vary in the trainer class. Thank you!