def range_train(): sample_sizes = [1e6] for size in sample_sizes: exp = Experiment("leagueoflegends", size, 75) model, losses, iterations = exp.regular_train() exp.predict(model, 1000, "s", True) plt.title("Losses for size " + str(size)) plt.plot(iterations, losses) plt.xlabel("Iteration") plt.ylabel("Loss") if not os.path.exists("plots/" + str(size) + ".png"): plt.savefig("plots/" + str(size) + ".png") else: count = 1 while os.path.exists("plots/" + str(size) + ".png"): count += 1 plt.savefig("plots/" + str(size) + ".png") print("#" * 80) print("FINAL LOSS FOR SIZE ", size, losses[len(losses) - 1]) print("#" * 80) plt.clf()
def custom(): filename = "../kelvin" exp = Experiment(subreddit=None, sample_size=0, percentile=0, custom=True, custom_file=filename) model, losses, iterations = exp.regular_train(epochs=5) exp.predict(model, 100, "s", False)
def single_complete_train(): exp = Experiment("leagueoflegends", None, 90) model, losses, iterations = exp.regular_train(epochs=1) exp.predict(model, 1000, "s", True) plt.title("Losses for size " + str(None)) plt.plot(iterations, losses) plt.xlabel("Iteration") plt.ylabel("Loss")