Skip to content

Conversation

@ailuntz
Copy link
Contributor

@ailuntz ailuntz commented Dec 17, 2025

Summary

This PR enhances the documentation to make the library more accessible to users, especially those facing environment setup challenges.

Changes

1. README.md Improvements

  • Docker Usage Section: Step-by-step guide for Docker setup
    • Build and run container
    • Copy datasets
    • Run zero-shot forecasting examples
  • Updated Requirements: Python 3.11 specification
  • Project Structure: Detailed directory layout explanation
  • Architecture Guide: E2E flow and layer descriptions
  • Learning Path: Recommended reading order for newcomers

2. README_zh.md (New)

  • Complete Chinese translation of the main README
  • Helps Chinese-speaking research community
  • Maintains consistency with English documentation
  • All sections translated including:
    • Installation guide
    • Usage examples
    • Project structure
    • Architecture explanation

Documentation Highlights

Docker Workflow Example

# Build and start
docker compose up -d --build

# Enter container
docker exec -it tslib bash

# Run zero-shot forecasting
python -u run.py \
  --task_name zero_shot_forecast \
  --model Moirai \
  --data ETTh1 \
  --seq_len 512 \
  --pred_len 96

Project Structure

Added comprehensive directory tree with descriptions for:

  • exp/ - Task pipelines
  • models/ - Model implementations
  • data_provider/ - Dataset loaders
  • layers/ - Reusable components
  • utils/ - Utility toolbox
  • scripts/ - Reproducible experiments

Architecture Overview

Documented the 5-layer architecture:

  • E2E flow from scripts to checkpoints
  • Experiment layer abstraction
  • Model and layer separation
  • Data provider patterns
  • Utility centralization

Benefits

  • ✅ Clearer onboarding for new users
  • ✅ Docker workflow documented
  • ✅ Better project understanding
  • ✅ Accessibility for Chinese researchers
  • ✅ Comprehensive learning guidance

Complementary PRs

This documentation enhancement complements:

Together, these PRs provide a complete solution for easy environment setup and improved developer experience.

Preview

The Chinese README maintains the same quality and completeness as the English version, ensuring consistency across languages.

This PR enhances the documentation to make the library more accessible
to users, especially those facing environment setup challenges.

Changes:
1. README.md improvements:
   - Add Docker usage section with step-by-step guide
   - Update Python version requirement to 3.11
   - Add detailed project structure documentation
   - Add architecture explanation section
   - Include zero-shot forecasting examples

2. README_zh.md (new):
   - Complete Chinese translation of README
   - Helps Chinese-speaking community
   - Maintains consistency with English version

Benefits:
- Clearer onboarding for new users
- Docker workflow documentation
- Better understanding of project architecture
- Accessibility for Chinese-speaking researchers
- Improved learning path recommendations

These documentation improvements complement the Docker support
and conditional imports PRs to provide a complete solution for
easy environment setup.
- Move data preparation section before installation
- Add detailed conda environment setup steps
- Add Mamba and Moirai dependency installation with CUDA notes
- Add quick test commands for all 5 tasks (1 epoch each)
- Reorganize sections: Installation, Docker Deployment, Train and Evaluate, Quick Test
- Improve navigation with clearer section hierarchy
- Update both English and Chinese documentation
- Update README.md and README_zh.md with latest improvements
- Refine section organization and formatting
- Enhance clarity and readability
@wuhaixu2016 wuhaixu2016 merged commit 2164135 into thuml:main Dec 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants