diff --git a/modules/deeplearning/quantile_regression.py b/modules/deeplearning/quantile_regression.py index 863f2f890c177797f3ba9f5974482e48de61de1a..a3bfaba6bd01f95dc311d9b49f0e71032e20f8d6 100644 --- a/modules/deeplearning/quantile_regression.py +++ b/modules/deeplearning/quantile_regression.py @@ -119,7 +119,7 @@ def run(num_points=1000, num_plot_pts=200): # Plot the results plt.figure(figsize=(8, 6)) - plt.scatter(X_test[:, 0], Y_test[:, 0], alpha=0.3, label="Test Data") + plt.scatter(X_test[::4, 0], Y_test[::4, 0], alpha=0.3, label="Test Data") plt.plot(X_range, predictions[0.05], label="Quantile 0.05", color='red') plt.plot(X_range, predictions[0.5], label="Quantile 0.5 (Median)", color='green') plt.plot(X_range, predictions[0.95], label="Quantile 0.95", color='blue')