64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import json
|
||
|
|
import collections
|
||
|
|
|
||
|
|
examples = [
|
||
|
|
{
|
||
|
|
"text": "icecream is cold",
|
||
|
|
"parsed": ("exists-property-with-value", 'icecream', 'cold'),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"text": "earth is a planet",
|
||
|
|
"parsed": ("pertenence-to-group", 'earth', 'planet'),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"text": "Green is a color",
|
||
|
|
"parsed": ("pertenence-to-group", 'green', 'color'),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"text": "airplanes do fly",
|
||
|
|
"parsed": ("has-capacity", 'airplane', 'fly')
|
||
|
|
}
|
||
|
|
]
|
||
|
|
|
||
|
|
knowledge_base = collections.defaultdict(lambda: collections.defaultdict(lambda: collections.defaultdict()))
|
||
|
|
|
||
|
|
|
||
|
|
def property_for_value(value):
|
||
|
|
if "cold":
|
||
|
|
return "temperature"
|
||
|
|
|
||
|
|
|
||
|
|
def exists_property_with_value(subj, value):
|
||
|
|
knowledge_base[subj][property_for_value(value)] = value
|
||
|
|
|
||
|
|
|
||
|
|
def pertenence_to_group(subj, group):
|
||
|
|
knowledge_base[subj]["group"] = group
|
||
|
|
|
||
|
|
|
||
|
|
def has_capacity(subj, capacity):
|
||
|
|
if "capacities" not in knowledge_base[subj]:
|
||
|
|
knowledge_base[subj]["capacities"] = []
|
||
|
|
knowledge_base[subj]["capacities"].append(capacity)
|
||
|
|
|
||
|
|
|
||
|
|
knowledge_ingestion = {
|
||
|
|
"exists-property-with-value": exists_property_with_value,
|
||
|
|
"pertenence-to-group": pertenence_to_group,
|
||
|
|
"has-capacity": has_capacity,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def ingest(example):
|
||
|
|
method = example['parsed'][0]
|
||
|
|
args = example['parsed'][1:]
|
||
|
|
knowledge_ingestion[method](*args)
|
||
|
|
|
||
|
|
|
||
|
|
for example in examples:
|
||
|
|
ingest(example)
|
||
|
|
|
||
|
|
print(json.dumps(knowledge_base, indent=4, sort_keys=True))
|