Official repository for the ICWSM '21 paper "More than meets the tie: Examining the Role of Interpersonal Relationships in Social Networks" (link) If using the package for research, please use the following citation:
@article{choi2021relationships,
title={More than Meets the Tie: Examining the Role of Interpersonal Relationships in Social Networks},
volume={15},
url={https://ojs.aaai.org/index.php/ICWSM/article/view/18045},
number={1},
journal={Proceedings of the International AAAI Conference on Web and Social Media},
author={Choi, Minje and Budak, Ceren and Romero, Daniel M. and Jurgens, David},
year={2021},
month={May},
pages={105-116} }
This code was tested under the following package dependencies. The package versions do not have to be exact.
- pytorch==1.10.0
- transformers==4.9.2
- tokenizers==0.10.1
- tweepy==3.10.0
- tqdm==4.49.0
- pandas==1.2.4
- scikit-learn==0.24.2
This model also requires the installation of BerTweet (link). Please keep this in mind if you plan to run the model in an offline environment where the huggingface models and tokenizers cannot be automatically installed.
We provide methods for inferring the relationships between any two Twitter users, provided that both of their tweet information is available by the Twitter API. You will need to have an application of the Twitter API (link).
Model trained on all features: [(link)](https://drive.google.com/file/d/1jrDiaSWJyrx6U08qlXg9kaWq9-vgYRzN/view?usp=share_link)
Model trained on all features except network features: [(link)](https://drive.google.com/file/d/1Z1536SEIT9cVk127FY6nvdaCPf84GSD5/view?usp=share_link)
Either the username (the name following @) or the user's ID number can be used, but they must be separated through a ','.
Examples:
username1,username2
user_id1,user_id2
user_id3,username3
The interactions will be stored in data/sample_outputs.json
.
You have to change the data as the same format as data/train-data-camera-ready.json.gz
.
- The file should be gzip-zipped
- Each line should be a json object that contains the interactions between the dyads
CUDA_VISIBLE_DEVICES=0 python run.py \
--do_predict --predict_batch_size=16 --accelerator gpu \
--use_public_mention --use_direct_mention --use_retweet \
--use_bio --use_activity --use_count --use_network \
--predict_data_file FILE_PATH_OF_INPUT_FILE \
--save_file=FILE_PATH_OF_OUTPUT_FILE_CONTAINING_INFERRED_SCORES \
--ckpt_path=FILE_PATH_OF_PRETRAINED_MODEL
** Note: if you do not have access to the features obtained from network interactions, you can use the model without network data which yields similar performance **
Unfortunately, we do not provide the datasets which we used for training the models as it contains conversations of real Twitter users. However, we are able to provide a dataset consisting of the user IDs and the relationship categories of the user dyads in Twitter. If interested, please send a request email to [[email protected]]. These are available at (link)
For questions, please contact the main author via [[email protected]]. Thanks!