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(""))