Python Transition.from_category Examples

Example #1

File: Models.py Project: jbuckman/shift-reduce-dependency-parser

	def scores(self, X_i, possible_transitions=None):
		probs = self.internal_model.decision_function(X_i)[0]
		ans = []
		for i in range(len(self.internal_model.classes_)):
			transition = Transition.from_category(self.internal_model.classes_[i], score=probs[i])
			if possible_transitions == None or transition in possible_transitions:
				ans.append((probs[i], transition))
		return ans

Example #2

File: Models.py Project: jbuckman/shift-reduce-dependency-parser

	def predict(self, stack, buff, arcs, labels, previous_transitions):
		features = self.extract_features(None, stack, buff, arcs, labels, previous_transitions)
		X_i = self.x_vectorizer.transform(features)
		prediction = self.internal_model.predict(X_i)
		return Transition.from_category(prediction)