|
@ -68,7 +68,7 @@ def standardization(dataset): |
|
|
|
|
|
|
|
|
print(bcolors.YELLOW + 'Running standardization' + bcolors.ENDC) |
|
|
print(bcolors.YELLOW + 'Running standardization' + bcolors.ENDC) |
|
|
# Standardization |
|
|
# Standardization |
|
|
scaledDataset = preprocessing.scale(dataset) |
|
|
scaledDataset = preprocessing.scale(np.float64(dataset)) |
|
|
|
|
|
|
|
|
return scaledDataset |
|
|
return scaledDataset |
|
|
|
|
|
|
|
@ -95,4 +95,4 @@ if __name__ == "__main__": |
|
|
print(bcolors.GREEN + 'Saving scaled results to file' + bcolors.ENDC) |
|
|
print(bcolors.GREEN + 'Saving scaled results to file' + bcolors.ENDC) |
|
|
datasetFrame = pd.DataFrame(scaledDataset, columns = featureKeys) |
|
|
datasetFrame = pd.DataFrame(scaledDataset, columns = featureKeys) |
|
|
datasetFrame = datasetFrame.assign(target=target) |
|
|
datasetFrame = datasetFrame.assign(target=target) |
|
|
datasetFrame.to_pickle("./dataset.pkl") |
|
|
datasetFrame.to_pickle("./dataset.pkl") |
|
|