run_exp.sh

# Running experiments for DP WGAN and DP Diffusion.

# Running epsilon = 2 for both WGAN and Diffusion on Appraise took ~1 hour.
# Haven't checked numbers, but port reconstruction for WGAN looks especially bad lmao.

DSET="appraise"
SYN_DATA_DIR="syn/${DSET}"

echo "Creating directory: ${SYN_DATA_DIR}"
mkdir -p "${SYN_DATA_DIR}"


RAW_DATA="data/nfiot_sample.csv"
PREPROC_DATA="data/nfiot_preproc.parquet"
# PREPROC_DATA="data/appraise_preproc.parquet"


# python prepare_data.py \
#     --data_path "${RAW_DATA}" \
#     --output_path "${PREPROC_DATA}" 


CUDA_VISIBLE_DEVICES=2 python run_diff_gan.py \
    --input_dataset "${DSET}" \
    --output_dir "${SYN_DATA_DIR}" \
    --models "DPDiffusion" 


# CUDA_VISIBLE_DEVICES=2 python ctgan2.py \
#     --input_dataset "appraise" \
#     --output_dir "${SYN_DATA_DIR}" \
#     --model "dpctgan"