Skip to content
Snippets Groups Projects
Commit 257cbc3b authored by Christian's avatar Christian
Browse files

Merge branch 'master' of https://github.com/ConvLab/ConvLab-3 into github_master

parents fe628952 ffb3dc42
No related branches found
No related tags found
No related merge requests found
...@@ -70,6 +70,10 @@ class VectorNodes(VectorBase): ...@@ -70,6 +70,10 @@ class VectorNodes(VectorBase):
if self.filter_state: if self.filter_state:
self.kg_info = self.filter_inactive_domains(domain_active_dict) self.kg_info = self.filter_inactive_domains(domain_active_dict)
# make sure kg is not empty
if len(self.kg_info) == 0:
self.add_user_greet()
if self.use_mask: if self.use_mask:
mask = self.get_mask(domain_active_dict, number_entities_dict) mask = self.get_mask(domain_active_dict, number_entities_dict)
for i in range(self.da_dim): for i in range(self.da_dim):
...@@ -170,3 +174,14 @@ class VectorNodes(VectorBase): ...@@ -170,3 +174,14 @@ class VectorNodes(VectorBase):
return kg_filtered return kg_filtered
def add_user_greet(self):
feature_type = 'user act'
da = ("general", "greet", "none", "none")
if da in self.opp2vec:
domain = da[0]
description = "user-" + "_".join(da)
value = 1.0
self.add_graph_node(domain, feature_type, description.lower(), value)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment