GitHub - lhysgithub/ORA: Source code for the paper "ORA: Job Runtime Prediction for High-Performance Computing Platforms Using the Online Retrieval-Augmented Language Model"

Source code for the paper "ORA: Job Runtime Prediction for High-Performance Computing Platforms Using the Online Retrieval-Augmented Language Model"

Get Started

1. Install Python 3.9. Follow the tutorial for setting up a local RAG environment using Langchain: [https://www.langchain.com.cn/docs/tutorials/local_rag/].
2. If using OpenAI's API, please fill in your OpenAI API Key in main.py.
3. Train and evaluate by running python main.py. The result records will be saved in the res directory.

The overview of ORA

Main Result