comparison demo.py @ 1:2082aeb1f1be tip

added demo and readme file

+import clausiepy as cp
+quote="Crows are feeding on rubbish at a garbage dump."
+
+# Extract clauses
+clauses = cp.clausie(quote)
+
+# Extract propositions
+propositions = cp.extract_propositions(clauses)
+
+queries = []
+
+# For every proposition, remove auxiliary verb and construct queries
+# (see paper)
+
+keys = ('subject', 'verb',  'indirect object',  'direct object', 'complement', 'adverb')
+
+queries = []
+
+for prop in propositions:
+
+    # Normal queries based on propositions with verbs
+    for L in range(len(keys), 1, -1):
+        chosen_keys = keys[:L]
+        propo = {}
+        for key in chosen_keys:
+            if key in prop:
+                propo[key] = prop[key]
+        p0 = cp.proposition_text(propo, chosen_keys)
+        prop_text = " ".join(" ".join([p1.text for p1 in p if p1.dep_ not in ['aux', 'det']]) for p in p0 if len(p) > 0)
+        if len(prop_text) > 0 and prop_text not in queries:
+            queries.append(prop_text)
+
+    # Subjects and objects independently
+    p0 = cp.proposition_text(prop, ['subject'])
+    prop_text = " ".join(" ".join([p1.text for p1 in p if p1.dep_ not in ['aux', 'det']]) for p in p0 if len(p) > 0)
+    if len(prop_text) > 0 and prop_text not in queries:
+        queries.append(prop_text)
+
+    p0 = cp.proposition_text(prop, ['indirect object'])
+    prop_text = " ".join(" ".join([p1.text for p1 in p if p1.dep_ not in ['aux', 'det', 'prep']]) for p in p0 if len(p) > 0)
+    if len(prop_text) > 0 and prop_text not in queries:
+        queries.append(prop_text)
+
+    p0 = cp.proposition_text(prop, ['direct object'])
+    prop_text = " ".join(" ".join([p1.text for p1 in p if p1.dep_ not in ['aux', 'det', 'prep']]) for p in p0 if len(p) > 0)
+    if len(prop_text) > 0 and prop_text not in queries:
+        queries.append(prop_text)        
+
+    # Adverb
+    p0 = cp.proposition_text(prop, ['adverb'])
+    prop_text = " ".join(" ".join([p1.text for p1 in p if p1.dep_ not in ['aux', 'det', 'prep']]) for p in p0 if len(p) > 0)
+    if len(prop_text) > 0 and prop_text not in queries:
+        queries.append(prop_text)        
+        
+
+print("Queries:")
+for query in queries:
+    print(query)