KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #typescript #developer-tools #claude-code #ai-agent #codex #deepseek-harness #ai-agents #cli #react #mcp

TH-GNN

★ 61 Open GitHub ↗

Code for TH-GNN (paper accepted by KDD2022)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# TH-GNN

Implementation of the KDD2022 paper, Company-as-Tribe: Company Risk Assessment On Tribe-Style Graph With Hierarchical Graph Neural Networks. [[Paper]](https://dl.acm.org/doi/10.1145/3534678.3539129)  [[Video]](https://dl.acm.org/action/downloadSupplement?doi=10.1145%2F3534678.3539129&file=KDD22-apfp1626.mp4)

TH-GNN (our proposed model) is a hierarchical graph neural network model for company financial risk assessment tasks. Each listed company with its investment graph is viewed as a tribe, which is more informative than  a single company.

![model_overview](https://raw.githubusercontent.com/wendongbi/TH-GNN/main/model_structure.png)

## Installation

### Requirements

* Linux with Python >= 3.7

* Pytorch >= 1.9.0

* DGL >= 0.7.0

* Scikit-learn >= 0.24.2

* Network >= 2.6.2

* Numpy >= 1.19.2

## Quick Start

### Dataset

Due to company privacy and copyright issues in the design of the datasets that we used, we do not disclose the company attributes and risk labels used in the paper.
* Company relationship data can be obtained through the paid API provided by Tianyancha (https://open.tianyancha.com/open/783). The above website also provides data samples in json format. Besides, node attributes are in csv format (company ID, feature 1, feature 2,...), which can be read through pandas. The node label comes from the wind database (https://www.wind.com.cn/mobile/EDB/zh.html) and is combined with expert annotations. The data involves corporate privacy and cannot be disclosed. Thank you for your understanding. For more information about data format and reading issues, please refer to the files: data_preprocess.py and dataset.py in the repository.

### Training

We provide an example for training the TH-GNN model in ./run.sh

```shell
python3 main.py \
 --gpu 0 \
 --local_num_layer 2 \
 --hidden 64 \
 --num_epoch 100 \
 --tribe_encoder_gnn gin \
 --lr 3e-3 \
 --weight_decay 5e-3 \
 --fusion_mode mlp \
 --path_x {path to node attribute file} \
 --path_y {path to node label file} \
 --path_tribe_files {path to tribe-graph files} \
 --path_tribe_order {path to tribe_graph order file}
```

TH-GNN

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README