# REPLACE WITH YOURS
datadir=/shared/share_mala/implicitbayes/dataset_files/MIND_data/large/
# from generate_text_labels.sh
Z_file=/shared/share_mala/implicitbayes/dataset_files/MIND_data/large/news_data_all_sentiment_and_formality_normalized.pt

export PYTHONPATH=$(pwd)/..

N=1000
D=20000
D_eval=10000
dimX=5
one_X_per_col=False
method='0111_logistic_withZ_zero_prodsign_last'

flip=True
Z_synthetic=False 
repeat_train_Z=1

python -m data_prep.generate_data_context --datadir $datadir --D $D --N $N --D_eval $D_eval --method $method --one_X_per_col $one_X_per_col --dimX $dimX --flip $flip --Z_file $Z_file --Z_synthetic $Z_synthetic --repeat_train_Z $repeat_train_Z 