<gh_stars>0
# Viz
import matplotlib.pyplot as plt


def plot_train_test(y, y_pred_train, y_test_pred):

    figure, ax = plt.subplots(figsize=(14, 4))

    line_obs = ax.plot(y, label='Observed')
    line_train = ax.plot(y_pred_train, label='y_train_pred')
    line_test = ax.plot(y_test_pred, label='y_test_pred')

    plt.title('Plot of Train and Test Set', fontdict={'fontsize': 16})

    plt.show()

# Viz
import matplotlib.pyplot as plt


def plot_train_test(y, y_pred_train, y_test_pred):

    figure, ax = plt.subplots(figsize=(14, 4))

    line_obs = ax.plot(y, label='Observed')
    line_train = ax.plot(y_pred_train, label='y_train_pred')
    line_test = ax.plot(y_test_pred, label='y_test_pred')

    plt.title('Plot of Train and Test Set', fontdict={'fontsize': 16})

    plt.show()