Additional if statement to solve the problem of 2 units

netrack / keras-metrics

Metrics for Keras. DEPRECATED since Keras 2.3.0

MIT License

165 stars 23 forks source link

Additional if statement to solve the problem of 2 units #4

Closed Avcu closed 6 years ago

Avcu commented 6 years ago

Since to_categorical function always put the true labels into second column when used with 2 classes, an additional if statement can be added in case of usage of more than one unit at the last layer.

Closes #3

Avcu commented 6 years ago

For an example code, kindly refer the link below https://colab.research.google.com/drive/1jeRnEJlqT82OT9Tibviq3JTo_8qOPnql#scrollTo=rErzh9evOiSY

ybubnov commented 6 years ago

@Avcu , thank you for submitting this PR!