Exemplo n.º 1
0
 def update_folder_paths(self):
     BaseModel.update_folder_paths(self)
     self.DTW_ROOT = os.path.join(self.ROOT, "dtw")
     self.DTW_DATA = os.path.join(self.DTW_ROOT, "dtw.npy")
     self.CLUSTER_ROOT = os.path.join(self.ROOT, "cluster")
     self.CLUSTER_DATA = os.path.join(self.CLUSTER_ROOT, "cluster.pickle")
     self.DICT_ROOT = os.path.join(self.ROOT, "dict")
     self.DICT_DATA = os.path.join(self.DICT_ROOT, "cluster.pickle")
Exemplo n.º 2
0
    def __init__(self, *args):
        BaseModel.__init__(self, *args)

        self.ALL = ["clean", "dtw", "cluster", "init", "sdict"]
        self.COMMANDS = self.ALL + ["pad", "find", "commands"]

        self.CLUSTER_MATRIX_SIZE = 1000 
        self.N_CLUSTERS = 50

        # one of 'single', 'complete', 'average', 'weighted'
        self.CLUSTERING_METHOD = "complete"

        self.STATE_MULTIPLIER = 5

        self.TRAIN_CORPORA = ["tomoe-ja-kanji"]
        self.EVAL_CORPORA = ["japanese-learner1", "japanese-native1"]
        self.ROOT = os.path.join("models", "strokehmm")
        self.update_folder_paths()
Exemplo n.º 3
0
    def clean(self):
        BaseModel.clean(self)

        for path in (self.DTW_ROOT, self.CLUSTER_ROOT, self.DICT_ROOT):
            if os.path.exists(path):
                shutil.rmtree(path)