Coverage for source/agent/strategies/classification_testable.py: 83%

1# agent/strategies/classification_testable.py 

2 

3# global imports 

4import pandas as pd 

5from abc import ABC, abstractmethod 

6 

7# local imports 

8 

9class ClassificationTestable(ABC): 

10 """ 

11 Implements an abstract base class for classification testable agents. 

12 """ 

13 

14 @abstractmethod 

15 def classify(self, data: pd.DataFrame) -> list[list[float]]: 

16 """ 

17 Classifies the input data using the trained model. 

18 

19 Parameters: 

20 data (pd.DataFrame): The input data to be classified. 

21 

22 Returns: 

23 (list[list[float]]): The predicted class probabilities for each input sample. 

24 """ 

25 

26 pass