示例#1
0
文件: main.py 项目: Superjom/TDA
class CentreDicBuilder:
    def __init__(self):
        self.dic = Dic()
        self.feature_extrator = FeatureExtrator()
        self.config = Config()

    def buildPath(self):
        basepath = self.config.get('path', 'source')
        return [basepath + path for path in os.listdir(basepath)]

    def run(self):
        #log
        open('CentreDicBuilder.log', 'w').write(' ')
        fi = open('CentreDicBuilder.log', 'a')
        for path in self.buildPath():
            c = open(path).read()
            self.feature_extrator.setSource(c)
            features = self.feature_extrator.getFeatures()
            for f in features:
                print f
                self.dic.add(f)
                fi.write(f + '\n')
        self.dic.done()
        self._tofile()
        fi.close()

    def _tofile(self, filename='features.dic'):
        self.dic.tofile(filename)
示例#2
0
文件: main.py 项目: Superjom/TDA
 def __init__(self):
     self.dic = Dic()
     self.feature_extrator = FeatureExtrator()
     self.config = Config()