# MSTNN

**Repository Path**: bearcatl/MSTNN

## Basic Information

- **Project Name**: MSTNN
- **Description**: No description available
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-11-23
- **Last Updated**: 2025-11-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## MSTNN

### 1. Environment & Installation Steps

python 3.8, pytorch, torch_geometric, numpy, pandas, sklearn and scipy.

$ pip install -r requirements.txt

### 2. Data

stock sequence data is under ./data/2013-01-01

hypergraph data file industry_NASDAQ.npy and industry_NYSE.npy is under ./training

## 3. Run

Execute the following python command to train MSTNN:

1) training on NSADQ:
   
   python train.py --market_name 'NASDAQ' --num_stocks 1026 --lr 0.0005  --dropout 0.1

2) training on NYSE:
   
   python train.py --market_name 'NYSE' --num_stocks 1737 --lr 0.001 --dropout 0.2