DataProcess.py 632 B

123456789101112131415161718192021
  1. import numpy as np
  2. import torch
  3. Data = np.loadtxt('./DataLib/Data.csv', delimiter=',')
  4. NumState = 13
  5. DataState = Data[:, :NumState]
  6. DataStateMax = np.max(DataState, axis=0)
  7. DataStateMin = np.min(DataState, axis=0)
  8. DataStateMaxMin = np.vstack([DataStateMax, DataStateMin])
  9. print(DataStateMaxMin)
  10. DataStateNor = 2*(DataState - DataStateMin) / (DataStateMax - DataStateMin) - 1
  11. np.savetxt('./DataLib/DataNor.csv', DataStateNor, delimiter=',')
  12. DataStateNor = torch.tensor(DataStateNor, dtype=torch.float32)
  13. torch.save(DataStateNor, './DataLib/Data.pt')
  14. np.savetxt('./DataLib/DataStateMaxMin.csv', DataStateMaxMin, delimiter=',')