e@0: #!/usr/bin/env python3
e@0: # -*- coding: utf-8 -*-
e@0: """
e@0: Created on Mon Apr 30 17:49:36 2018
e@0: 
e@0: @author: Emmanouil Theofanis Chourdakis
e@0: """
e@0: 
e@0: from sklearn.svm import LinearSVC
e@0: from sklearn.feature_extraction import DictVectorizer
e@0: 
e@0: class RelModel(LinearSVC):
e@0:     ## TODO: Add more LinearSVC parameters here
e@0:     def __init__(self):
e@0:         super(RelModel, self).__init__()
e@0:         self.dv = DictVectorizer()
e@0:         
e@0:     def fit(self, X, y, sample_weight=None):
e@0:         
e@0:         # Transform data and save transformer
e@0:         x = self.dv.fit_transform(X)
e@0:         
e@0:         return super(RelModel, self).fit(x, y, sample_weight)
e@0:     
e@0:     def predict(self, X):
e@0:         # Transform data with transformer
e@0:         x = self.dv.transform(X)
e@0:         
e@0:         return super(RelModel, self).predict(x)