graph_model_results.py

h_predictions, *_ = learn.get_preds(DatasetType.Test)
#h_predictions = h_predictions.numpy()

v_predictions, *_ = learn.get_preds(DatasetType.Valid)
#h_predictions = h_predictions.numpy()

predictions, *_ = learn.get_preds(DatasetType.Train)
#predictions = predictions.numpy()

issue = "foxnews.com_russia"

results = time_series[[issue]][-52:]
results.columns = ["actual"]

results["predicted"] = (np.concatenate((predictions, v_predictions, h_predictions), axis=None))
results["m"] = results.actual[:len(predictions)].mean()


ax = results[["actual", "m"]].rename(columns={"m":"mean"}).plot(figsize=(12,8))
results["predicted"][:len(predictions)].plot(ax=ax, color="orangered", label="Train", title=issue)
results["predicted"][len(predictions)-1:-len(h_predictions)].plot(ax=ax, color="red", label="Validation")
results["predicted"][-len(h_predictions)-1:].plot(ax=ax, color="darkred", label="Test")