RS2002/

WiCount

	WiCount




	
	Sample Usage

To use this dataset with the CSI-BERT2 code, first clone the repository: git clone https://github.com/RS2002/CSI-BERT2 cd CSI-BERT2

Then you can use the provided scripts for pre-training, fine-tuning, and inference. Replace with the path to the WiCount dataset downloaded from Hugging Face.

python pretrain.py --GAN --data_path

If you do not want to use the discriminator, you can omit the --GAN flag.… See the full description on the dataset page: https://huggingface.co/datasets/RS2002/WiCount.