mirror of
https://github.com/ANL-CEEESA/MIPLearn.git
synced 2025-12-07 09:58:51 -06:00
20 lines
403 B
Python
20 lines
403 B
Python
from abc import ABC, abstractmethod
|
|
|
|
import numpy as np
|
|
|
|
from miplearn.h5 import H5File
|
|
|
|
|
|
class FeaturesExtractor(ABC):
|
|
@abstractmethod
|
|
def get_instance_features(self, h5: H5File) -> np.ndarray:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_var_features(self, h5: H5File) -> np.ndarray:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_constr_features(self, h5: H5File) -> np.ndarray:
|
|
pass
|