main.py

from src.data import load_data
from src.model import RNNModel
from src.train import train_model
from src.generate import generate_text
import matplotlib.pyplot as plt

data, char2idx, idx2char, vocab_size = load_data()

print("Total characters:", len(data))
print("Unique characters:", vocab_size)
model = RNNModel(vocab_size)
print(model)
model, losses = train_model(data, vocab_size)
text = generate_text(model, "To be", char2idx, idx2char)
print("Temp 0.5:\n", generate_text(model, "To be", char2idx, idx2char, temperature=0.5))
print("\nTemp 1.0:\n", generate_text(model, "To be", char2idx, idx2char, temperature=1.0))
print("\nTemp 1.2:\n", generate_text(model, "To be", char2idx, idx2char, temperature=1.2))
print(text)
print(generate_text(model, "Hello", char2idx, idx2char))
print(generate_text(model, "The king", char2idx, idx2char))
print(generate_text(model, "Love", char2idx, idx2char))

plt.plot(losses)
plt.title("Training Loss")
plt.show()