def generate(model,
gen_count,
temperature,
weight_adjust={
'.': 2,
'?': 2,
'!': 2,
',': 2
}):
generations = []
for _ in range(gen_count):
generation = custom_generate(model,
temperature=temperature,
weight_adjustments=weight_adjust,
include_stop_token=True)
# If generation was able to finish (as opposed to being manually stopped),
# and if tweet contains letters (is not just series of symbols)
if generation[-3:] == '<s>' and ParseTools.contains_letters(
generation[:-3]):
# Clip stop character and save tweet
generation = generation[:-3]
generations.append(generation)
return generations
def test_contains_letters(self):
self.assertTrue(ParseTools.contains_letters("String with letters"))
self.assertFalse(ParseTools.contains_letters(".!.?"))
self.assertFalse(ParseTools.contains_letters("."))
self.assertFalse(ParseTools.contains_letters("//"))
self.assertFalse(ParseTools.contains_letters(""))
