|国家预印本平台
首页|A Binary Classification Social Network Dataset for Graph Machine Learning

A Binary Classification Social Network Dataset for Graph Machine Learning

A Binary Classification Social Network Dataset for Graph Machine Learning

来源:Arxiv_logoArxiv
英文摘要

Social networks have a vast range of applications with graphs. The available benchmark datasets are citation, co-occurrence, e-commerce networks, etc, with classes ranging from 3 to 15. However, there is no benchmark classification social network dataset for graph machine learning. This paper fills the gap and presents the Binary Classification Social Network Dataset (\textit{BiSND}), designed for graph machine learning applications to predict binary classes. We present the BiSND in \textit{tabular and graph} formats to verify its robustness across classical and advanced machine learning. We employ a diverse set of classifiers, including four traditional machine learning algorithms (Decision Trees, K-Nearest Neighbour, Random Forest, XGBoost), one Deep Neural Network (multi-layer perceptrons), one Graph Neural Network (Graph Convolutional Network), and three state-of-the-art Graph Contrastive Learning methods (BGRL, GRACE, DAENS). Our findings reveal that BiSND is suitable for classification tasks, with F1-scores ranging from 67.66 to 70.15, indicating promising avenues for future enhancements.

Jinglong Li、Adnan Ali、AlMotasem Bellah Al Ajlouni、Huanhuan Chen

计算技术、计算机技术

Jinglong Li,Adnan Ali,AlMotasem Bellah Al Ajlouni,Huanhuan Chen.A Binary Classification Social Network Dataset for Graph Machine Learning[EB/OL].(2025-03-04)[2025-05-28].https://arxiv.org/abs/2503.02397.点此复制

评论