Update README.md
Browse files
README.md
CHANGED
@@ -7,5 +7,21 @@ datasets:
|
|
7 |
|
8 |
# DARA: Decomposition-Alignment-Reasoning Autonomous Language Agent for Question Answering over Knowledge Graphs
|
9 |
|
10 |
-
|
11 |
-
This model is a fine-tuned semantic parsing LLM agent for KGQA.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
# DARA: Decomposition-Alignment-Reasoning Autonomous Language Agent for Question Answering over Knowledge Graphs
|
9 |
|
10 |
+
## Model Information
|
11 |
+
This model is a fine-tuned semantic parsing LLM agent for KGQA. We fine-tune the llama-2-13B on our curated reasoning trajectory https://huggingface.co/datasets/UKPLab/dara.
|
12 |
+
|
13 |
+
Model Usage
|
14 |
+
from transformers import AutoModelForCausalLM
|
15 |
+
|
16 |
+
model = AutoModelForCausalLM.from_pretrained(
|
17 |
+
"UKPLab/dara-llama-2-13b",
|
18 |
+
torch_dtype=torch.float16,
|
19 |
+
device_map="auto",
|
20 |
+
cache_dir = "cache"
|
21 |
+
)
|
22 |
+
For more information, please check the repository https://github.com/UKPLab/acl2024-DARA
|
23 |
+
|
24 |
+
Hyperparameters
|
25 |
+
Learning rate: 2e-5
|
26 |
+
Batch size: 4
|
27 |
+
Training epochs: 10
|