Memo for tojima

featureExtract.py

def exstractFeature1(text):
    return [len(text)]

class HasChar(object):
    def __init__(self, ch):
        self.ch = ch
    def __call__(self, text):
        return [0 if text.find(self.ch)<0 else 1]


class FeatureExtractor(object):
    def __init__(self, features):
        self._features = features

    def __call__(self, text):
        new_feature = []
        for f in self._features:
            new_feature.extend(f(text))
        return new_feature


d = {
 'Length': exstractFeature1,
 'HasA': HasChar('A'),
 'HasB': HasChar('B'),
}

features = FeatureExtractor(d.values())
print features('HAS') #-> [3, 1, 0]

実行結果