Skip to content
Snippets Groups Projects
Commit e839d910 authored by Carel van Niekerk's avatar Carel van Niekerk :computer:
Browse files

Bug fixes

parent 29781af6
No related branches found
No related tags found
No related merge requests found
...@@ -31,7 +31,7 @@ from convlab.dst.setsumbt.modeling.bert_nbt import BertSetSUMBT ...@@ -31,7 +31,7 @@ from convlab.dst.setsumbt.modeling.bert_nbt import BertSetSUMBT
from convlab.dst.setsumbt.modeling.roberta_nbt import RobertaSetSUMBT from convlab.dst.setsumbt.modeling.roberta_nbt import RobertaSetSUMBT
from convlab.dst.setsumbt.unified_format_data import unified_format from convlab.dst.setsumbt.unified_format_data import unified_format
from convlab.dst.setsumbt.modeling import training from convlab.dst.setsumbt.modeling import training
from convlab.dst.setsumbt.multiwoz import ontology as embeddings from convlab.dst.setsumbt.unified_format_data.dataset import ontology as embeddings
from convlab.dst.setsumbt.utils import get_args, update_args from convlab.dst.setsumbt.utils import get_args, update_args
# from convlab.dst.setsumbt.modeling import ensemble_utils # from convlab.dst.setsumbt.modeling import ensemble_utils
......
...@@ -10,20 +10,11 @@ from transformers import (BertModel, BertConfig, BertTokenizer, ...@@ -10,20 +10,11 @@ from transformers import (BertModel, BertConfig, BertTokenizer,
from convlab.dst.setsumbt.modeling import (RobertaSetSUMBT, from convlab.dst.setsumbt.modeling import (RobertaSetSUMBT,
BertSetSUMBT) BertSetSUMBT)
<<<<<<<< HEAD:convlab/dst/setsumbt/multiwoz/Tracker.py
from convlab.dst.dst import DST from convlab.dst.dst import DST
from convlab.util.multiwoz.state import default_state from convlab.util.multiwoz.state import default_state
from convlab.util.multiwoz.multiwoz_slot_trans import REF_SYS_DA, REF_USR_DA from convlab.util.multiwoz.multiwoz_slot_trans import REF_SYS_DA, REF_USR_DA
from convlab.dst.rule.multiwoz import normalize_value from convlab.dst.rule.multiwoz import normalize_value
from convlab.util.custom_util import model_downloader from convlab.util.custom_util import model_downloader
========
from convlab2.dst.dst import DST
from convlab2.util.multiwoz.state import default_state
from convlab2.util.multiwoz.multiwoz_slot_trans import REF_SYS_DA, REF_USR_DA
from convlab2.dst.rule.multiwoz import normalize_value
from convlab2.util.custom_util import model_downloader
from convlab2.dst.setsumbt.modeling.training import set_ontology_embeddings
>>>>>>>> setsumbt_unifiedformat:convlab2/dst/setsumbt/unified_format_data/Tracker.py
USE_CUDA = torch.cuda.is_available() USE_CUDA = torch.cuda.is_available()
......
from convlab2.dst.setsumbt.unified_format_data.dataset import unified_format, ontology from convlab.dst.setsumbt.unified_format_data.dataset import unified_format, ontology
from convlab2.dst.setsumbt.unified_format_data.Tracker import SetSUMBTTracker from convlab.dst.setsumbt.unified_format_data.Tracker import SetSUMBTTracker
\ No newline at end of file \ No newline at end of file
#!/usr/bin/env python3
"""__init__.py.py:Description"""
__author__ = 'Carel van Niekerk'
__version__ = '0.1'
__date__ = '29.06.22'
# Load Packages
if __name__ == '__main__':
#run
...@@ -19,7 +19,7 @@ import torch ...@@ -19,7 +19,7 @@ import torch
from torch.utils.data import Dataset, DataLoader, RandomSampler, SequentialSampler from torch.utils.data import Dataset, DataLoader, RandomSampler, SequentialSampler
from copy import deepcopy from copy import deepcopy
from convlab2.dst.setsumbt.unified_format_data.dataset.utils import (load_dataset, get_ontology_slots, from convlab.dst.setsumbt.unified_format_data.dataset.utils import (load_dataset, get_ontology_slots,
ontology_add_values, get_values_from_data, ontology_add_requestable_slots, ontology_add_values, get_values_from_data, ontology_add_requestable_slots,
get_requestable_slots, load_dst_data, extract_dialogues) get_requestable_slots, load_dst_data, extract_dialogues)
......
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
# limitations under the License. # limitations under the License.
"""Convlab3 Unified dataset data processing utilities""" """Convlab3 Unified dataset data processing utilities"""
from convlab2.util import load_dataset, load_ontology, load_dst_data, load_nlu_data from convlab.util import load_dataset, load_ontology, load_dst_data, load_nlu_data
# MultiWOZ specific label map to avoid duplication and typos in values # MultiWOZ specific label map to avoid duplication and typos in values
VALUE_MAP = {'guesthouse': 'guest house', 'belfry': 'belfray', '-': ' ', '&': 'and', 'b and b': 'bed and breakfast', VALUE_MAP = {'guesthouse': 'guest house', 'belfry': 'belfray', '-': ' ', '&': 'and', 'b and b': 'bed and breakfast',
......
...@@ -29,7 +29,7 @@ setup( ...@@ -29,7 +29,7 @@ setup(
'tabulate', 'tabulate',
'python-Levenshtein', 'python-Levenshtein',
'requests', 'requests',
'numpy>=1.22', 'numpy',
'nltk', 'nltk',
'scipy', 'scipy',
'tensorboard', 'tensorboard',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment