def getRegularFeatures(data, isTrainData): data = performRegularFeatureEngineering(data, isTrainData) # splitting data into X and Y if 'Category' in data.columns.values: yData = data.Category data = data.drop(['Category'], 1) else: yData = [] xData = data.values dataReader.serializeObject(data.columns.values,"data\\misc\\columns.csv") print "Features used {}".format(data.columns.values) return xData,yData