def test_feature_correlation_integrated_mutual_info_regression(self):
"""
Test FeatureCorrelation visualizer with mutual information regression
"""
viz = FeatureCorrelation(method="mutual_info-regression")
viz.fit(self.X, self.y, random_state=23456)
viz.finalize()
self.assert_images_similar(viz)
def test_feature_correlation_integrated_pearson(self):
"""
Test FeatureCorrelation visualizer with pearson correlation
coefficient
"""
viz = FeatureCorrelation()
viz.fit(self.X, self.y)
viz.finalize()
self.assert_images_similar(viz)
def test_feature_correlation_integrated_mutual_info_classification(self):
"""
Test FeatureCorrelation visualizer with mutual information
on wine dataset (classification)
"""
data = datasets.load_wine()
X, y = data["data"], data["target"]
viz = FeatureCorrelation(method="mutual_info-classification")
viz.fit(X, y, random_state=12345)
viz.finalize()
self.assert_images_similar(viz)