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