diff options
author | ZJaume <jzaragoza@prompsit.com> | 2022-03-25 12:23:08 +0300 |
---|---|---|
committer | ZJaume <jzaragoza@prompsit.com> | 2022-03-25 12:23:08 +0300 |
commit | d7f853c6f40ca4645f57e05bd4830227c9f3e484 (patch) | |
tree | 4247f623a24e5225e6cb2dfd0bb2cb1e2ec4802a | |
parent | 284b0bb196ceeeeb7261dab9920c90d2fe593754 (diff) | |
parent | fe01ea476dd22cc20feab8a517c6ea52ef091241 (diff) |
Merge branch 'master' into gensim
-rw-r--r-- | bicleaner_ai/models.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/bicleaner_ai/models.py b/bicleaner_ai/models.py index 646e03a..45c31ce 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -261,6 +261,7 @@ class BaseModel(ModelInterface): deps = {'FScore': FScore, 'MatthewsCorrCoef': MatthewsCorrCoef, 'TokenAndPositionEmbedding': TokenAndPositionEmbedding, + 'K': tf.keras.backend, } # Try loading the whole model |