Source code for autogl.module.feature.selectors.base

from ..base import BaseFeatureAtom
import numpy as np


[docs]class BaseSelector(BaseFeatureAtom): def __init__(self, data_t="np", multigraph=False,**kwargs): super(BaseSelector, self).__init__(data_t=data_t, multigraph=multigraph,**kwargs) self._sel = None def _transform(self, data): if self._sel is not None: data.x = data.x[:, self._sel] return data