Store the dataset after imputation

pfnet-research / TabCSDI

A code for the NeurIPS 2022 Table Representation Learning Workshop paper: "Diffusion models for missing value imputation in tabular data"

MIT License

42 stars 4 forks source link

Store the dataset after imputation #4

Closed kuiyuanshao closed 6 months ago

kuiyuanshao commented 6 months ago

Hi there,

I am wondering how to store the dataset after the imputation? I tried to use the model.evaluate() and store the "samples" object, but got very large and werid values compared to original data.

Thank you in advance!

joshua-shuhan commented 6 months ago

Hi, thanks for your interest in our work. Have you tried to uncomment codes here?

kuiyuanshao commented 6 months ago

Hi Joshua,

I guess I have figured out how to save the dataset, thank you for your quick reply.