feat: modify modality and optimizer

pnpxai/__init__.py

@@ -1,10 +1,4 @@
 from pnpxai.core.detector import detect_model_architecture
 from pnpxai.core.recommender import XaiRecommender
-from pnpxai.core.experiment import (
-    Experiment,
-    AutoExplanation,
-    AutoExplanationForImageClassification,
-    AutoExplanationForTextClassification,
-    AutoExplanationForVisualQuestionAnswering,
-    AutoExplanationForTSClassification,
-)    
+from pnpxai.core.experiment import Experiment, AutoExplanation
+from pnpxai.core.modality import Modality

pnpxai/core/__init__.py

@@ -1,10 +1,5 @@
 from pnpxai.core.detector import detect_model_architecture
 from pnpxai.core.recommender import XaiRecommender
-from pnpxai.core.experiment import (
-    Experiment,
-    AutoExplanation,
-    AutoExplanationForImageClassification,
-    AutoExplanationForTextClassification,
-    AutoExplanationForVisualQuestionAnswering,
-    AutoExplanationForTSClassification,
-)
+from pnpxai.core.experiment import Experiment, AutoExplanation
+from pnpxai.core.utils import ModelWrapper
+from pnpxai.core.modality import Modality

pnpxai/core/_types.py

@@ -13,6 +13,7 @@
 Task = Literal["classification"]
 ExplanationType = Literal["attribution"]
 
+
 class ConfigKeys(Enum):
     EXPLAINERS = 'explainers'
     METRICS = 'metrics'

pnpxai/core/detector/__init__.py

@@ -1 +1,6 @@
-from pnpxai.core.detector.detector import detect_model_architecture, symbolic_trace, extract_graph_data
+from pnpxai.core.detector.detector import (
+    detect_model_architecture,
+    symbolic_trace,
+    extract_graph_data,
+    detect_data_modality,
+)