def predictAll(self, reader): """Prediction function for all features from a certain reader.""" feats, labels = featurizer(reader, self.featureFuncs, self.client) featMat = self.dictVectorizer.transform(feats) return self.classifier.predict(featMat)
def train(self): feats, labels = featurizer(sick_train_reader, self.featureFuncs, self.client) featMat = self.dictVectorizer.fit_transform(feats) self.classifier.fit(featMat, labels)