Views
No views yet
cacifer_rnn_dreamer.keras – Keras model filechar_to_idx.json – mapping from character → indexidx_to_char.json – mapping from index → character1import json
2import numpy as np
3from tensorflow.keras.models import load_model
4
5model = load_model("cacifer_rnn_dreamer.keras", compile=False)
6char_to_idx = json.load(open("char_to_idx.json"))
7idx_to_char = {int(k): v for k, v in json.load(open("idx_to_char.json")).items()}
8seq_len = 60
9
10def sample_char(probs, temperature=0.7):
11 probs = np.asarray(probs).astype("float64")
12 probs = np.log(probs + 1e-8) / temperature
13 probs = np.exp(probs) / np.sum(probs)
14 return np.random.choice(len(probs), p=probs)
15
16def generate(seed, length=400, temperature=0.7):
17 x = seed[-seq_len:]
18 for _ in range(length):
19 x_idx = np.array([[char_to_idx.get(c, 0) for c in x]])
20 preds = model.predict(x_idx, verbose=0)[0]
21 next_idx = sample_char(preds, temperature)
22 x += idx_to_char[next_idx]
23 return x
24
25seed = "I am Cacifer, a cat, listening to the bells and the breathing of the guests."
26print(generate(seed, length=400, temperature=0.7))
27
28---