Forgot tensorflow import - Githubissues

mkocabas / focal-loss-keras

Focal Loss implementation in Keras

MIT License

328 stars 74 forks source link

Forgot tensorflow import #15

Closed yassineAlouini closed 5 years ago

yassineAlouini commented 6 years ago

In the https://github.com/mkocabas/focal-loss-keras/blob/master/focal_loss.py file, you forgot to import tensorflow: import tensorflow as tf.

chengdabao commented 5 years ago

u r right

mkocabas commented 5 years ago

https://github.com/mkocabas/focal-loss-keras/commit/c40cacff60bda15b02bd83ea3a1b2be739d56117 should solve the issue.