diff --git a/README.md b/README.md index c07806d..ed5867e 100644 --- a/README.md +++ b/README.md @@ -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.