diff --git a/classifier/preprocessing/data_preprocessing.py b/classifier/preprocessing/data_preprocessing.py index cbd6344..5fa9541 100644 --- a/classifier/preprocessing/data_preprocessing.py +++ b/classifier/preprocessing/data_preprocessing.py @@ -68,7 +68,7 @@ def standardization(dataset): print(bcolors.YELLOW + 'Running standardization' + bcolors.ENDC) # Standardization - scaledDataset = preprocessing.scale(dataset) + scaledDataset = preprocessing.scale(np.float64(dataset)) return scaledDataset @@ -95,4 +95,4 @@ if __name__ == "__main__": print(bcolors.GREEN + 'Saving scaled results to file' + bcolors.ENDC) datasetFrame = pd.DataFrame(scaledDataset, columns = featureKeys) datasetFrame = datasetFrame.assign(target=target) - datasetFrame.to_pickle("./dataset.pkl") \ No newline at end of file + datasetFrame.to_pickle("./dataset.pkl")