import matplotlib.pyplot as plt
from sklearn.feature_selection import f_regression, mutual_info_regression
X = np.random.rand(1000, 3)
y = X[:, 0] + np.sin(6 * np.pi * X[:, 1]) + 0.1 * np.random.randn(1000)
f_test, _ = f_regression(X, y)
mi = mutual_info_regression(X, y)
plt.figure(figsize=(15, 5))
plt.xlabel("$x_{}quot;.format(i + 1), fontsize=14)
plt.ylabel("$yquot;, fontsize=14)
plt.title("F-test={:.2f}, MI={:.2f}".format(f_test[i], mi[i]),