Unable to run last step of `elephas/examples/ml_mlp_classification.py`

danielenricocahall / elephas

Distributed Deep learning with Keras & Spark

MIT License

17 stars 5 forks source link

Unable to run last step of `elephas/examples/ml_mlp_classification.py` #4

Closed Yoseph-Zuskin closed 2 years ago

Yoseph-Zuskin commented 2 years ago

I get the following error when trying to run the second last line of elephas/examples/ml_mlp_classification.py:

TypeError: precision() missing 1 required positional argument: 'label'

danielenricocahall commented 2 years ago

Hello @Yoseph-Zuskin, yes it looks like the API changed plus the output format from the model during classification changed a few releases ago. I just made the update to fix the example and also illustrate how to use MulticlassMetrics. Thank you!