This commit is contained in:
wassname
2024-05-11 11:01:49 +08:00
parent 0c5d523d74
commit 3f66da5544
+36 -30
View File
@@ -46,36 +46,42 @@ prompt = "Generate a young person's information based on the following schema:"
jsonformer = Jsonformer(model, tokenizer, json_schema, prompt, temperature=0)
generated_data = jsonformer()
generated_data = {'name': 'John',
'age_probs': [{'prob': 0.62353515625, 'choice': '10'},
{'prob': 0.276611328125, 'choice': '12'},
{'prob': 0.05364990234375, 'choice': '20'},
{'prob': 0.0257415771484375, 'choice': '11'},
{'prob': 0.0047607421875, 'choice': '15'},
{'prob': 0.004688262939453125, 'choice': '16'},
{'prob': 0.002910614013671875, 'choice': '18'},
{'prob': 0.0024127960205078125, 'choice': '13'},
{'prob': 0.0015821456909179688, 'choice': '14'},
{'prob': 0.0013532638549804688, 'choice': '23'},
{'prob': 0.0012521743774414062, 'choice': '21'},
{'prob': 0.00042247772216796875, 'choice': '17'},
{'prob': 0.0003342628479003906, 'choice': '22'},
{'prob': 0.0002484321594238281, 'choice': '19'},
{'prob': 0.0001995563507080078, 'choice': '25'},
{'prob': 4.851818084716797e-05, 'choice': '24'},
{'prob': 3.30805778503418e-05, 'choice': '26'},
{'prob': 2.6404857635498047e-05, 'choice': '28'},
{'prob': 1.728534698486328e-05, 'choice': '27'},
{'prob': 2.9802322387695312e-06, 'choice': '29'}],
'age_wmean': 17.32853078842163,
'is_student_probs': [{'prob': 0.8173828125, 'choice': 'true'},
{'prob': 0.182373046875, 'choice': 'false'}],
'is_student': False,
'age': 17,
'unit_time': 0.5,
'courses': ['CS101'],
'trim': None,
'color': 'white'}
generated_data = {
"age_probs": [
{"prob": 0.94091796875, "choice": "10"},
{"prob": 0.033233642578125, "choice": "20"},
{"prob": 0.0122222900390625, "choice": "12"},
{"prob": 0.00412750244140625, "choice": "21"},
{"prob": 0.0028362274169921875, "choice": "16"},
{"prob": 0.0018453598022460938, "choice": "15"},
{"prob": 0.00113677978515625, "choice": "11"},
{"prob": 0.0011110305786132812, "choice": "18"},
{"prob": 0.0005083084106445312, "choice": "25"},
{"prob": 0.0004558563232421875, "choice": "23"},
{"prob": 0.0002498626708984375, "choice": "14"},
{"prob": 0.00023281574249267578, "choice": "13"},
{"prob": 0.0002238750457763672, "choice": "22"},
{"prob": 0.00018131732940673828, "choice": "26"},
{"prob": 0.0001690387725830078, "choice": "24"},
{"prob": 0.00012552738189697266, "choice": "19"},
{"prob": 7.796287536621094e-05, "choice": "27"},
{"prob": 7.265806198120117e-05, "choice": "28"},
{"prob": 4.106760025024414e-05, "choice": "17"},
{"prob": 2.5033950805664062e-06, "choice": "29"},
],
"age_wmean": 17.816404402256012,
"is_student_probs": [
{"prob": 0.974609375, "choice": "true"},
{"prob": 0.025177001953125, "choice": "false"},
],
"is_student": False,
"name": "John",
"age": 17,
"unit_time": 0.5,
"courses": ["C++"],
"trim": None,
"color": "white",
}
```
The original [README](https://github.com/1rgs/jsonformer) is includes below.