har1 commited on
Commit
81bdfab
1 Parent(s): fb411a5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -10
README.md CHANGED
@@ -33,7 +33,8 @@ The model is used to generate clinical notes from doctor-patient conversation da
33
  - When the input data is composed of very minimal character tokens or if input is very large it starts to hallucinate.
34
 
35
 
36
- ### TEST DATA Sample For Inference (More given in `test.txt`)
 
37
  "Doctor: Hi there, I love that dress, very pretty!
38
  Patient: Thank you for complementing a seventy-two-year-old patient.
39
  Doctor: No, I mean it, seriously. Okay, so you were admitted here in May two thousand nine. You have a history of hypertension, and on June eighteenth two thousand nine you had bad abdominal pain diarrhea and cramps.
@@ -48,21 +49,23 @@ Doctor: Okay. Anything else any other symptoms?
48
  Patient: Actually no. Everything's well.
49
  Doctor: Great.
50
  Patient: Yeah."
51
-
52
 
53
  ## Training and evaluation data
54
 
55
- It achieves the following results on the evaluation set:
56
- - Loss: 0.1562
57
- - Rouge1: 54.3238
58
- - Rouge2: 34.2678
59
- - Rougel: 46.5847
60
- - Rougelsum: 51.2214
61
- - Gen Len: 77.04
 
 
62
 
63
  ## Training procedure
64
 
65
- The model was trained on 1201 training samples and 100 validation samples of the modified [MTS-Dialog]()
66
 
67
  ### Training hyperparameters
68
 
 
33
  - When the input data is composed of very minimal character tokens or if input is very large it starts to hallucinate.
34
 
35
 
36
+ ### TEST DATA Sample For Inference (More given in [`test.txt`](https://huggingface.co/har1/HealthScribe-Clinical_Note_Generator/blob/main/test.txt))
37
+ ```
38
  "Doctor: Hi there, I love that dress, very pretty!
39
  Patient: Thank you for complementing a seventy-two-year-old patient.
40
  Doctor: No, I mean it, seriously. Okay, so you were admitted here in May two thousand nine. You have a history of hypertension, and on June eighteenth two thousand nine you had bad abdominal pain diarrhea and cramps.
 
49
  Patient: Actually no. Everything's well.
50
  Doctor: Great.
51
  Patient: Yeah."
52
+ ```
53
 
54
  ## Training and evaluation data
55
 
56
+ The model achieves the following results on the evaluation set:
57
+
58
+ - **Loss:** 0.1562
59
+ - **Rouge1:** 54.3238
60
+ - **Rouge2:** 34.2678
61
+ - **Rougel:** 46.5847
62
+ - **Rougelsum:** 51.2214
63
+ - **Generation Length:** 77.04
64
+
65
 
66
  ## Training procedure
67
 
68
+ The model was trained on 1201 training samples and 100 validation samples of the modified [MTS-Dialog](https://huggingface.co/datasets/har1/MTS_Dialogue-Clinical_Note)
69
 
70
  ### Training hyperparameters
71