Fix AttributeError: 'ProgbarLogger' object has no attribute 'params' for tf2.3.0

Author: hankcs

hanlp/components/parsers/biaffine_parser_tf.py

@@ -102,6 +102,8 @@ def train_loop(self, trn_data, dev_data, epochs, num_examples,
         c: tf.keras.callbacks.Callback = None
         metric = self._build_metrics()
         for c in callbacks:
+            if not hasattr(c, 'params'):
+                c.params = dict()
             c.params['epochs'] = epochs
             c.params['trn_data'] = trn_data
             c.params['metrics'] = ['loss'] + self.config.metrics

plugins/hanlp_demo/hanlp_demo/zh/tf/train_ctb5_dep.py

@@ -9,17 +9,17 @@
 cdroot()
 save_dir = 'data/model/dep/biaffine_ctb'
 parser = BiaffineDependencyParserTF()
-# parser.fit(CTB5_DEP_TRAIN, CTB5_DEP_DEV, save_dir,
-#            pretrained_embed={'class_name': 'HanLP>Word2VecEmbedding',
-#                              'config': {
-#                                  'trainable': False,
-#                                  'embeddings_initializer': 'zero',
-#                                  'filepath': CTB5_FASTTEXT_300_CN,
-#                                  'expand_vocab': True,
-#                                  'lowercase': True,
-#                                  'normalize': True,
-#                              }},
-#            )
+parser.fit(CTB5_DEP_TRAIN, CTB5_DEP_DEV, save_dir,
+           pretrained_embed={'class_name': 'HanLP>Word2VecEmbedding',
+                             'config': {
+                                 'trainable': False,
+                                 'embeddings_initializer': 'zero',
+                                 'filepath': CTB5_FASTTEXT_300_CN,
+                                 'expand_vocab': True,
+                                 'lowercase': True,
+                                 'normalize': True,
+                             }},
+           )
 parser.load(save_dir)
 sentence = [('中国', 'NR'), ('批准', 'VV'), ('设立', 'VV'), ('外商', 'NN'), ('投资', 'NN'), ('企业', 'NN'), ('逾', 'VV'),
             ('三十万', 'CD'), ('家', 'M')]