from sklearn.neural_network import MLPClassifier
from MLModel import MLModel

class SVMClassifierModel(MLModel):
    def __init__(self, model):
        self.model = model

    def train(self, X, y):
        self.model.fit(X, y)

    def predict(self, X):
        return self.model.predict(X.values)