Browse Source

avoid sklern.scale() error for converting float32 to float64

master
Frank Blanning 6 years ago
parent
commit
1a5b548e3e
  1. 4
      classifier/preprocessing/data_preprocessing.py

4
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")
datasetFrame.to_pickle("./dataset.pkl")

Loading…
Cancel
Save