Skip to content

Latest commit

 

History

History
44 lines (31 loc) · 1.9 KB

README.md

File metadata and controls

44 lines (31 loc) · 1.9 KB

📈 (Dataset Available) -- ChatGPT Informed Graph Neural Network for Stock Movement Prediction 📈

  • This repository houses the datasets/resources used in our paper ChatGPT Informed Graph Neural Network for Stock Movement Prediction.
  • This research introduces a novel framework that leverages ChatGPT's graph inference capabilities to enhance Graph Neural Networks (GNN).
  • Our model shows superior performance in both (1) stock movement prediction and (2) portfolio construction.
  • Dive in to find the datasets, code samples, and more!
  • 📄 Paper: [Link to the paper]

🚀 Get Started

The data can be found in the Data folder, which contains two files:

  • ticker_train_data.json: This file holds the data utilized for training and validation of our model.
  • ticker_test_data.json: This file contains the data used for model evaluation.

To load the data, you can start with 4 lines of code:

import pandas as pd
import json

train_data = pd.read_json('./Data/ticker_train_data.json')
test_data = pd.read_json('./Data/ticker_test_data.json')

The Affected Companies column provides two key insights:

  • Companies that ChatGPT predicts will be influenced by the financial news.
  • The sentiment indicating the nature of the impact on these companies (e.g., positive or negative).

For a deeper exploration of the data, please feel free to check the data_checking.ipynb.

🔗 Citation:

We encourage collaboration and use of this dataset for further advancements in stock prediction using deep learning. If you find this resource useful, kindly cite our paper. Happy researching!

@article{chen2023chatgpt,
  title={ChatGPT Informed Graph Neural Network for Stock Movement Prediction},
  author={Chen, Zihan and Zheng, Lei Nico and Lu, Cheng and Yuan, Jialu and Zhu, Di},
  journal={arXiv preprint arXiv:2306.03763},
  year={2023}
}