import inspect
import numpy as np
import pandas as pd
from fedbiomed.common.training_plans import SKLearnTrainingPlan
from fedbiomed.common.data import DataManager
from sklearn.linear_model import SGDClassifier
from fedbiomed.common.training_plans import FedPerceptron
class PerceptronTrainingPlan(FedPerceptron):
    def training_data(self):
        NUMBER_COLS = 20
        dataset = pd.read_csv(self.dataset_path,header=None,delimiter=',')
        X = dataset.iloc[:,0:NUMBER_COLS].values
        y = dataset.iloc[:,NUMBER_COLS]
        return DataManager(dataset=X,target=y.values,  shuffle=True)
