feat: add correlation plot functions to init

bhklab · Dec 16, 2024 · 0f9f331 · 0f9f331
1 parent 6370597
commit 0f9f331
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/readii/analyze/__init__.py b/src/readii/analyze/__init__.py
@@ -1,10 +1,14 @@
 """ Module to perform analysis on READII outputs """
 
 from .correlation import getFeatureCorrelations, getVerticalSelfCorrelations, getHorizontalSelfCorrelations, getCrossCorrelationMatrix
+from.plot_correlation import plotCorrelationHeatmap, plotCorrelationHistogram
+
 
 __all__ = [
     'getFeatureCorrelations',
     'getVerticalSelfCorrelations',
     'getHorizontalSelfCorrelations',
-    'getCrossCorrelationMatrix'
+    'getCrossCorrelationMatrix',
+    'plotCorrelationHeatmap',
+    'plotCorrelationHistogram'
 ]