diff --git a/src/sdqc_check/classification/classification_model.py b/src/sdqc_check/classification/classification_model.py index 84a9093..7cec835 100644 --- a/src/sdqc_check/classification/classification_model.py +++ b/src/sdqc_check/classification/classification_model.py @@ -1,4 +1,5 @@ import warnings +from abc import ABC from typing import Optional, Dict, Union, List, Tuple import pandas as pd @@ -21,7 +22,7 @@ warnings.filterwarnings('ignore') -class ClassificationModel: +class ClassificationModel(ABC): """ A class for training and evaluating binary classification models to distinguish between real and synthetic data.