Adjust the weight of nadya.jp edges

commonsense · Jan 17, 2018 · dfca826 · dfca826
1 parent 6f74067
commit dfca826
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/conceptnet5/readers/conceptnet4.py b/conceptnet5/readers/conceptnet4.py
@@ -319,6 +319,9 @@ def build_sources(parts_dict, preposition_fix=False):
 
 # TODO: this doesn't need to be a class
 class CN4Builder(object):
+    def __init__(self, weight=1.):
+        self.weight = weight
+
     def handle_assertion(self, parts_dict):
         """
         Process one assertion from ConceptNet 4, which appears in the input
@@ -374,7 +377,7 @@ def handle_assertion(self, parts_dict):
                     rel=relation, start=start, end=end,
                     dataset=dataset, license=Licenses.cc_attribution,
                     sources=[source_dict], surfaceText=frame_text,
-                    weight=weight
+                    weight=weight * self.weight
                 )
 
     def transform_file(self, input_filename, output_file):

diff --git a/conceptnet5/readers/nadya.py b/conceptnet5/readers/nadya.py
@@ -90,7 +90,7 @@ def handle_line(line, builder):
 
 
 def handle_file(input_filename, output_file):
-    builder = CN4Builder()
+    builder = CN4Builder(weight=0.1)
     out = MsgpackStreamWriter(output_file)
     for line in open(input_filename, encoding='utf-8'):
         # Get a line from the file