X_train, X_test = tf.split(tf.random.shuffle(X, seed=42), num_or_size_splits=[40, 10])
def plot_predictions(train_data = X_train, train_labels = y_train, test_data = X_test, test_labels = y_test, predictions = y_pred):
""" Plots training data, testing_data """
plt.figure(figsize=(10, 7))
plt.scatter(train_data, train_labels, c="blue", label='Training data')
plt.scatter(test_data, test_labels, c="green", label="Testing data")
plt.scatter(test_data, predictions, c="red", label="Predictions")
plt.legend();
Common regression evaluation metrics