Update README.md
Browse files
README.md
CHANGED
@@ -1,39 +1,46 @@
|
|
1 |
---
|
2 |
-
license:
|
|
|
3 |
---
|
4 |
|
5 |
-
#
|
6 |
|
7 |
<!-- Provide a quick summary of what the model is/does. -->
|
8 |
|
9 |
-
dragon-
|
10 |
|
11 |
DRAGON models have been fine-tuned with the specific objective of fact-based question-answering over complex business and legal documents with an emphasis on reducing hallucinations and providing short, clear answers for workflow automation.
|
12 |
|
|
|
13 |
### Benchmark Tests
|
14 |
|
15 |
Evaluated against the benchmark test: [RAG-Instruct-Benchmark-Tester](https://www.huggingface.co/datasets/llmware/rag_instruct_benchmark_tester)
|
16 |
-
|
17 |
-
|
18 |
-
--**Accuracy Score**: **
|
19 |
-
--Not Found Classification:
|
20 |
-
--Boolean:
|
21 |
-
--Math/Logic:
|
22 |
-
--Complex Questions (1-5):
|
23 |
-
--Summarization Quality (1-5):
|
24 |
--Hallucinations: No hallucinations observed in test runs.
|
25 |
|
26 |
-
For test run results (and good indicator of target use cases), please see the files ("core_rag_test" and "answer_sheet" in this repo).
|
|
|
|
|
|
|
|
|
|
|
27 |
|
28 |
### Model Description
|
29 |
|
30 |
<!-- Provide a longer summary of what this model is. -->
|
31 |
|
32 |
- **Developed by:** llmware
|
33 |
-
- **Model type:**
|
34 |
- **Language(s) (NLP):** English
|
35 |
-
- **License:**
|
36 |
-
- **Finetuned from model:**
|
37 |
|
38 |
|
39 |
### Direct Use
|
@@ -59,8 +66,8 @@ Any model can provide inaccurate or incomplete information, and should be used i
|
|
59 |
The fastest way to get started with dRAGon is through direct import in transformers:
|
60 |
|
61 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
62 |
-
tokenizer = AutoTokenizer.from_pretrained("dragon-
|
63 |
-
model = AutoModelForCausalLM.from_pretrained("dragon-
|
64 |
|
65 |
Please refer to the generation_test .py files in the Files repository, which includes 200 samples and script to test the model. The **generation_test_llmware_script.py** includes built-in llmware capabilities for fact-checking, as well as easy integration with document parsing and actual retrieval to swap out the test set for RAG workflow consisting of business documents.
|
66 |
|
|
|
1 |
---
|
2 |
+
license: apache-2.0
|
3 |
+
inference: false
|
4 |
---
|
5 |
|
6 |
+
# DRAGON-MISTRAL-0.3
|
7 |
|
8 |
<!-- Provide a quick summary of what the model is/does. -->
|
9 |
|
10 |
+
dragon-mistral-0.3 is part of the dRAGon ("Delivering RAG On ...") model series, RAG-instruct trained on top of a Mistral 7b (0.3) base model.
|
11 |
|
12 |
DRAGON models have been fine-tuned with the specific objective of fact-based question-answering over complex business and legal documents with an emphasis on reducing hallucinations and providing short, clear answers for workflow automation.
|
13 |
|
14 |
+
|
15 |
### Benchmark Tests
|
16 |
|
17 |
Evaluated against the benchmark test: [RAG-Instruct-Benchmark-Tester](https://www.huggingface.co/datasets/llmware/rag_instruct_benchmark_tester)
|
18 |
+
1 Test Run (with temperature = 0.0 and sample = False) with 1 point for correct answer, 0.5 point for partial correct or blank / NF, 0.0 points for incorrect, and -1 points for hallucinations.
|
19 |
+
|
20 |
+
--**Accuracy Score**: **99.5** correct out of 100
|
21 |
+
--Not Found Classification: 95.0%
|
22 |
+
--Boolean: 82.5%
|
23 |
+
--Math/Logic: 67.5%
|
24 |
+
--Complex Questions (1-5): 4 (Above Average - multiple-choice, causal)
|
25 |
+
--Summarization Quality (1-5): 4 (Above Average)
|
26 |
--Hallucinations: No hallucinations observed in test runs.
|
27 |
|
28 |
+
For test run results (and good indicator of target use cases), please see the files ("core_rag_test" and "answer_sheet" in this repo).
|
29 |
+
|
30 |
+
Please note that these test results were achieved using the 4_K_M quantized version of this model - [dragon-mistral-0.3-gguf](https://wwww.huggingface.co/llmware/dragon-mistral-0.3-gguf).
|
31 |
+
|
32 |
+
Note: compare results with [dragon-mistral-7b](https://www.huggingface.co/llmware/dragon-mistral-7b-v0).
|
33 |
+
|
34 |
|
35 |
### Model Description
|
36 |
|
37 |
<!-- Provide a longer summary of what this model is. -->
|
38 |
|
39 |
- **Developed by:** llmware
|
40 |
+
- **Model type:** mistral
|
41 |
- **Language(s) (NLP):** English
|
42 |
+
- **License:** Apache 2.0
|
43 |
+
- **Finetuned from model:** Mistral-7b-base-0.3
|
44 |
|
45 |
|
46 |
### Direct Use
|
|
|
66 |
The fastest way to get started with dRAGon is through direct import in transformers:
|
67 |
|
68 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
69 |
+
tokenizer = AutoTokenizer.from_pretrained("dragon-mistral-0.3")
|
70 |
+
model = AutoModelForCausalLM.from_pretrained("dragon-mistral-0.3")
|
71 |
|
72 |
Please refer to the generation_test .py files in the Files repository, which includes 200 samples and script to test the model. The **generation_test_llmware_script.py** includes built-in llmware capabilities for fact-checking, as well as easy integration with document parsing and actual retrieval to swap out the test set for RAG workflow consisting of business documents.
|
73 |
|