1.16.4

Joostb · Jul 9, 2022 · 5067397 · 5067397
1 parent 194b333
commit 5067397
Show file tree

Hide file tree

Showing 11 changed files with 82 additions and 561 deletions.
diff --git a/README.MD b/README.MD
@@ -10,6 +10,7 @@
 [![total views per week](https://raw.githubusercontent.com/yangheng95/PyABSA/traffic/total_views_per_week.svg)](https://github.com/yangheng95/PyABSA/tree/traffic#-total-traffic-data-badge)
 [![total clones](https://raw.githubusercontent.com/yangheng95/PyABSA/traffic/total_clones.svg)](https://github.com/yangheng95/PyABSA/tree/traffic#-total-traffic-data-badge)
 [![total clones per week](https://raw.githubusercontent.com/yangheng95/PyABSA/traffic/total_clones_per_week.svg)](https://github.com/yangheng95/PyABSA/tree/traffic#-total-traffic-data-badge)
+[![Coverage Status](https://coveralls.io/repos/github/yangheng95/PyABSA/badge.svg?branch=release)](https://coveralls.io/github/yangheng95/PyABSA?branch=release)
 
 [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/back-to-reality-leveraging-pattern-driven/aspect-based-sentiment-analysis-on-semeval)](https://paperswithcode.com/sota/aspect-based-sentiment-analysis-on-semeval?p=back-to-reality-leveraging-pattern-driven)
 

diff --git a/basic_test/run_atepc_pretrain_test.py b/basic_test/run_atepc_pretrain_test.py
@@ -45,7 +45,7 @@
 ]
 
 # # for dataset in ABSADatasetList():
-for dataset in ABSADatasetList()[:1]:
+for dataset in ABSADatasetList()[:1]+[ABSADatasetList.MAMS]:
     for model in ATEPCModelList():
         config = ATEPCConfigManager.get_atepc_config_english()
         cuda.empty_cache()
@@ -55,6 +55,7 @@
         config.evaluate_begin = 0
         config.max_seq_len = 10
         config.log_step = -1
+        config.ate_loss_weight = 5
         config.show_metric = -1
         aspect_extractor = Trainer(config=config,
                                    dataset=dataset,