import csv

from AutoEncoderModel import AutoEncoder
import torch

model = torch.load('./ModelLib/models/model_nlat_100_nhid_128_lr_0.001.pt', map_location="cpu")

Data = torch.load('./DataLib/Data.pt')

input = Data
input_pre = model(input)
latent = model.encoder(input)

input_pre = input_pre.detach().numpy()
latent = latent.detach().numpy()
with open('./DataLib/DataPre.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(input_pre)

with open('./DataLib/LatentPre.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(latent)