123456789101112131415161718192021 |
- import numpy as np
- import torch
- Data = np.loadtxt('./DataLib/Data.csv', delimiter=',')
- NumState = 13
- DataState = Data[:, :NumState]
- DataStateMax = np.max(DataState, axis=0)
- DataStateMin = np.min(DataState, axis=0)
- DataStateMaxMin = np.vstack([DataStateMax, DataStateMin])
- print(DataStateMaxMin)
- DataStateNor = 2*(DataState - DataStateMin) / (DataStateMax - DataStateMin) - 1
- np.savetxt('./DataLib/DataNor.csv', DataStateNor, delimiter=',')
- DataStateNor = torch.tensor(DataStateNor, dtype=torch.float32)
- torch.save(DataStateNor, './DataLib/Data.pt')
- np.savetxt('./DataLib/DataStateMaxMin.csv', DataStateMaxMin, delimiter=',')
|