question of compute_accuracy about moon

[18811449050]

I would like to know what this method(compute_accuracy) does? centralized learning？Model warm up？

[QinbinLi]

Hi @18811449050 ,

This method computes the prediction accuracy of the input model on the input dataset. There is no training on the model inside compute_accuracy.

[18811449050]

If it is a regression task, how to change the loss?

[QinbinLi]

Currently, MOON is designed for the classification task. For regression task, you may need to change the cross-entropy loss (e.g., line 130 of main.py) to loss for regression (e.g., MSELoss). You also need to change the model architecture.