python 一个基于双面映射生成语言句子的小型原型。

Posted 2021-05-09

import random


class EventGen(object):

	def __init__(self, gen_len=3):
		self.objs = {}
		self.actions = {}
		self.pattern = ""
		self.gen_len = gen_len

	def set_pattern(self, pattern):
		self.pattern = pattern

	def set_len(self, length):
		self.gen_len = length

	def add_object(self, name, *childs):
		if name not in self.objs:
			self.objs[name] = []
		for elem in childs:
			self.objs[name].append(elem)

	def add_action(self, name, *childs):
		if name not in self.actions:
			self.actions[name] = []
		for elem in childs:
			self.actions[name].append(elem)

	def rand_obj(self):
		return random.choice(self.objs.keys())

	def rand_act(self):
		return random.choice(self.actions.keys())


	def gen_obj_act_obj(self):
		picked_obj = self.rand_obj()
		sub_act = random.choice(self.objs[picked_obj])
		term_obj = random.choice(self.actions[sub_act])
		return picked_obj + " " + sub_act + " " + term_obj + "."

"""
jweinstein-MBP-71683:startup jweinstein$ python -i gen.py
>>> test = EventGen()
>>> test
<__main__.EventGen object at 0x1034ba0d0>
>>> test.add_object('An apple', 'shines', 'looks')
>>> test.objs
{'An apple': ['shines', 'looks']}
>>> test.add_action('shines', 'bright', 'bold', 'boldly')
>>> test.add_action('looks', 'red', 'blue', 'reddish')
>>> test.gen_obj_act_obj()
'An apple shines bold.'
>>> test.gen_obj_act_obj()
'An apple looks red.'
>>> test.gen_obj_act_obj()
'An apple shines bold.'
>>> test.gen_obj_act_obj()
'An apple looks red.'
>>> test.gen_obj_act_obj()
'An apple shines boldly.'
>>> test.gen_obj_act_obj()
'An apple looks reddish.'
>>> exit()
"""