GNNRecom/gnnrec/config.py

11 lines
195 B
Python
Raw Permalink Normal View History

2021-11-16 07:04:52 +00:00
from pathlib import Path
# 项目根目录
BASE_DIR = Path(__file__).resolve().parent.parent
# 数据集目录
DATA_DIR = BASE_DIR / 'data'
# 模型保存目录
MODEL_DIR = BASE_DIR / 'model'