Hugging Face's logo

Spaces:
prabalGaur
/
Capstone
Running on Zero

App Files Community
prabalGaur commited on
Commit
fa00833
·
verified ·
1 Parent(s): 57f04b3

Upload results.ipynb with huggingface_hub

Browse files
Files changed (1) hide show
  1. results.ipynb +964 -0
results.ipynb ADDED
@@ -0,0 +1,964 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cells": [
3
+ {
4
+ "cell_type": "code",
5
+ "execution_count": 3,
6
+ "id": "a86fd346",
7
+ "metadata": {},
8
+ "outputs": [
9
+ {
10
+ "data": {
11
+ "application/vnd.plotly.v1+json": {
12
+ "config": {
13
+ "plotlyServerURL": "https://plot.ly"
14
+ },
15
+ "data": [
16
+ {
17
+ "marker": {
18
+ "color": [
19
+ "gray",
20
+ "gray",
21
+ "gray",
22
+ "gray",
23
+ "gray",
24
+ "black",
25
+ "orange",
26
+ "slateblue",
27
+ "slateblue",
28
+ "slateblue",
29
+ "slateblue",
30
+ "slateblue",
31
+ "skyblue",
32
+ "orange",
33
+ "darkred",
34
+ "red",
35
+ "red",
36
+ "green",
37
+ "purple"
38
+ ]
39
+ },
40
+ "type": "bar",
41
+ "x": [
42
+ "Constant",
43
+ "Linear Regression",
44
+ "NLP + LR",
45
+ "Random Forest",
46
+ "XGBoost",
47
+ "Human (Ed)",
48
+ "Neural Network",
49
+ "GPT 4.1 Nano",
50
+ "Grok 4.1 Fast",
51
+ "Gemini 3 Pro",
52
+ "Claude 4.5 Sonnet",
53
+ "GPT 5.1",
54
+ "GPT 4.1 Nano (Fine-tuned)",
55
+ "Deep Neural Network",
56
+ "Base Llama 3.2 4 bit",
57
+ "Fine-tuned Lite",
58
+ "Fine-tuned Full",
59
+ "GPT 5.1 RAG",
60
+ "Ensemble"
61
+ ],
62
+ "y": [
63
+ 106.18,
64
+ 101.56,
65
+ 76.81,
66
+ 72.28,
67
+ 68.23,
68
+ 87.62,
69
+ 63.97,
70
+ 62.51,
71
+ 57.62,
72
+ 50.54,
73
+ 47.1,
74
+ 44.74,
75
+ 75.91,
76
+ 46.49,
77
+ 110.72,
78
+ 65.4,
79
+ 39.85,
80
+ 30.19,
81
+ 29.9
82
+ ]
83
+ }
84
+ ],
85
+ "layout": {
86
+ "height": 800,
87
+ "template": {
88
+ "data": {
89
+ "bar": [
90
+ {
91
+ "error_x": {
92
+ "color": "#2a3f5f"
93
+ },
94
+ "error_y": {
95
+ "color": "#2a3f5f"
96
+ },
97
+ "marker": {
98
+ "line": {
99
+ "color": "#E5ECF6",
100
+ "width": 0.5
101
+ },
102
+ "pattern": {
103
+ "fillmode": "overlay",
104
+ "size": 10,
105
+ "solidity": 0.2
106
+ }
107
+ },
108
+ "type": "bar"
109
+ }
110
+ ],
111
+ "barpolar": [
112
+ {
113
+ "marker": {
114
+ "line": {
115
+ "color": "#E5ECF6",
116
+ "width": 0.5
117
+ },
118
+ "pattern": {
119
+ "fillmode": "overlay",
120
+ "size": 10,
121
+ "solidity": 0.2
122
+ }
123
+ },
124
+ "type": "barpolar"
125
+ }
126
+ ],
127
+ "carpet": [
128
+ {
129
+ "aaxis": {
130
+ "endlinecolor": "#2a3f5f",
131
+ "gridcolor": "white",
132
+ "linecolor": "white",
133
+ "minorgridcolor": "white",
134
+ "startlinecolor": "#2a3f5f"
135
+ },
136
+ "baxis": {
137
+ "endlinecolor": "#2a3f5f",
138
+ "gridcolor": "white",
139
+ "linecolor": "white",
140
+ "minorgridcolor": "white",
141
+ "startlinecolor": "#2a3f5f"
142
+ },
143
+ "type": "carpet"
144
+ }
145
+ ],
146
+ "choropleth": [
147
+ {
148
+ "colorbar": {
149
+ "outlinewidth": 0,
150
+ "ticks": ""
151
+ },
152
+ "type": "choropleth"
153
+ }
154
+ ],
155
+ "contour": [
156
+ {
157
+ "colorbar": {
158
+ "outlinewidth": 0,
159
+ "ticks": ""
160
+ },
161
+ "colorscale": [
162
+ [
163
+ 0,
164
+ "#0d0887"
165
+ ],
166
+ [
167
+ 0.1111111111111111,
168
+ "#46039f"
169
+ ],
170
+ [
171
+ 0.2222222222222222,
172
+ "#7201a8"
173
+ ],
174
+ [
175
+ 0.3333333333333333,
176
+ "#9c179e"
177
+ ],
178
+ [
179
+ 0.4444444444444444,
180
+ "#bd3786"
181
+ ],
182
+ [
183
+ 0.5555555555555556,
184
+ "#d8576b"
185
+ ],
186
+ [
187
+ 0.6666666666666666,
188
+ "#ed7953"
189
+ ],
190
+ [
191
+ 0.7777777777777778,
192
+ "#fb9f3a"
193
+ ],
194
+ [
195
+ 0.8888888888888888,
196
+ "#fdca26"
197
+ ],
198
+ [
199
+ 1,
200
+ "#f0f921"
201
+ ]
202
+ ],
203
+ "type": "contour"
204
+ }
205
+ ],
206
+ "contourcarpet": [
207
+ {
208
+ "colorbar": {
209
+ "outlinewidth": 0,
210
+ "ticks": ""
211
+ },
212
+ "type": "contourcarpet"
213
+ }
214
+ ],
215
+ "heatmap": [
216
+ {
217
+ "colorbar": {
218
+ "outlinewidth": 0,
219
+ "ticks": ""
220
+ },
221
+ "colorscale": [
222
+ [
223
+ 0,
224
+ "#0d0887"
225
+ ],
226
+ [
227
+ 0.1111111111111111,
228
+ "#46039f"
229
+ ],
230
+ [
231
+ 0.2222222222222222,
232
+ "#7201a8"
233
+ ],
234
+ [
235
+ 0.3333333333333333,
236
+ "#9c179e"
237
+ ],
238
+ [
239
+ 0.4444444444444444,
240
+ "#bd3786"
241
+ ],
242
+ [
243
+ 0.5555555555555556,
244
+ "#d8576b"
245
+ ],
246
+ [
247
+ 0.6666666666666666,
248
+ "#ed7953"
249
+ ],
250
+ [
251
+ 0.7777777777777778,
252
+ "#fb9f3a"
253
+ ],
254
+ [
255
+ 0.8888888888888888,
256
+ "#fdca26"
257
+ ],
258
+ [
259
+ 1,
260
+ "#f0f921"
261
+ ]
262
+ ],
263
+ "type": "heatmap"
264
+ }
265
+ ],
266
+ "histogram": [
267
+ {
268
+ "marker": {
269
+ "pattern": {
270
+ "fillmode": "overlay",
271
+ "size": 10,
272
+ "solidity": 0.2
273
+ }
274
+ },
275
+ "type": "histogram"
276
+ }
277
+ ],
278
+ "histogram2d": [
279
+ {
280
+ "colorbar": {
281
+ "outlinewidth": 0,
282
+ "ticks": ""
283
+ },
284
+ "colorscale": [
285
+ [
286
+ 0,
287
+ "#0d0887"
288
+ ],
289
+ [
290
+ 0.1111111111111111,
291
+ "#46039f"
292
+ ],
293
+ [
294
+ 0.2222222222222222,
295
+ "#7201a8"
296
+ ],
297
+ [
298
+ 0.3333333333333333,
299
+ "#9c179e"
300
+ ],
301
+ [
302
+ 0.4444444444444444,
303
+ "#bd3786"
304
+ ],
305
+ [
306
+ 0.5555555555555556,
307
+ "#d8576b"
308
+ ],
309
+ [
310
+ 0.6666666666666666,
311
+ "#ed7953"
312
+ ],
313
+ [
314
+ 0.7777777777777778,
315
+ "#fb9f3a"
316
+ ],
317
+ [
318
+ 0.8888888888888888,
319
+ "#fdca26"
320
+ ],
321
+ [
322
+ 1,
323
+ "#f0f921"
324
+ ]
325
+ ],
326
+ "type": "histogram2d"
327
+ }
328
+ ],
329
+ "histogram2dcontour": [
330
+ {
331
+ "colorbar": {
332
+ "outlinewidth": 0,
333
+ "ticks": ""
334
+ },
335
+ "colorscale": [
336
+ [
337
+ 0,
338
+ "#0d0887"
339
+ ],
340
+ [
341
+ 0.1111111111111111,
342
+ "#46039f"
343
+ ],
344
+ [
345
+ 0.2222222222222222,
346
+ "#7201a8"
347
+ ],
348
+ [
349
+ 0.3333333333333333,
350
+ "#9c179e"
351
+ ],
352
+ [
353
+ 0.4444444444444444,
354
+ "#bd3786"
355
+ ],
356
+ [
357
+ 0.5555555555555556,
358
+ "#d8576b"
359
+ ],
360
+ [
361
+ 0.6666666666666666,
362
+ "#ed7953"
363
+ ],
364
+ [
365
+ 0.7777777777777778,
366
+ "#fb9f3a"
367
+ ],
368
+ [
369
+ 0.8888888888888888,
370
+ "#fdca26"
371
+ ],
372
+ [
373
+ 1,
374
+ "#f0f921"
375
+ ]
376
+ ],
377
+ "type": "histogram2dcontour"
378
+ }
379
+ ],
380
+ "mesh3d": [
381
+ {
382
+ "colorbar": {
383
+ "outlinewidth": 0,
384
+ "ticks": ""
385
+ },
386
+ "type": "mesh3d"
387
+ }
388
+ ],
389
+ "parcoords": [
390
+ {
391
+ "line": {
392
+ "colorbar": {
393
+ "outlinewidth": 0,
394
+ "ticks": ""
395
+ }
396
+ },
397
+ "type": "parcoords"
398
+ }
399
+ ],
400
+ "pie": [
401
+ {
402
+ "automargin": true,
403
+ "type": "pie"
404
+ }
405
+ ],
406
+ "scatter": [
407
+ {
408
+ "fillpattern": {
409
+ "fillmode": "overlay",
410
+ "size": 10,
411
+ "solidity": 0.2
412
+ },
413
+ "type": "scatter"
414
+ }
415
+ ],
416
+ "scatter3d": [
417
+ {
418
+ "line": {
419
+ "colorbar": {
420
+ "outlinewidth": 0,
421
+ "ticks": ""
422
+ }
423
+ },
424
+ "marker": {
425
+ "colorbar": {
426
+ "outlinewidth": 0,
427
+ "ticks": ""
428
+ }
429
+ },
430
+ "type": "scatter3d"
431
+ }
432
+ ],
433
+ "scattercarpet": [
434
+ {
435
+ "marker": {
436
+ "colorbar": {
437
+ "outlinewidth": 0,
438
+ "ticks": ""
439
+ }
440
+ },
441
+ "type": "scattercarpet"
442
+ }
443
+ ],
444
+ "scattergeo": [
445
+ {
446
+ "marker": {
447
+ "colorbar": {
448
+ "outlinewidth": 0,
449
+ "ticks": ""
450
+ }
451
+ },
452
+ "type": "scattergeo"
453
+ }
454
+ ],
455
+ "scattergl": [
456
+ {
457
+ "marker": {
458
+ "colorbar": {
459
+ "outlinewidth": 0,
460
+ "ticks": ""
461
+ }
462
+ },
463
+ "type": "scattergl"
464
+ }
465
+ ],
466
+ "scattermap": [
467
+ {
468
+ "marker": {
469
+ "colorbar": {
470
+ "outlinewidth": 0,
471
+ "ticks": ""
472
+ }
473
+ },
474
+ "type": "scattermap"
475
+ }
476
+ ],
477
+ "scattermapbox": [
478
+ {
479
+ "marker": {
480
+ "colorbar": {
481
+ "outlinewidth": 0,
482
+ "ticks": ""
483
+ }
484
+ },
485
+ "type": "scattermapbox"
486
+ }
487
+ ],
488
+ "scatterpolar": [
489
+ {
490
+ "marker": {
491
+ "colorbar": {
492
+ "outlinewidth": 0,
493
+ "ticks": ""
494
+ }
495
+ },
496
+ "type": "scatterpolar"
497
+ }
498
+ ],
499
+ "scatterpolargl": [
500
+ {
501
+ "marker": {
502
+ "colorbar": {
503
+ "outlinewidth": 0,
504
+ "ticks": ""
505
+ }
506
+ },
507
+ "type": "scatterpolargl"
508
+ }
509
+ ],
510
+ "scatterternary": [
511
+ {
512
+ "marker": {
513
+ "colorbar": {
514
+ "outlinewidth": 0,
515
+ "ticks": ""
516
+ }
517
+ },
518
+ "type": "scatterternary"
519
+ }
520
+ ],
521
+ "surface": [
522
+ {
523
+ "colorbar": {
524
+ "outlinewidth": 0,
525
+ "ticks": ""
526
+ },
527
+ "colorscale": [
528
+ [
529
+ 0,
530
+ "#0d0887"
531
+ ],
532
+ [
533
+ 0.1111111111111111,
534
+ "#46039f"
535
+ ],
536
+ [
537
+ 0.2222222222222222,
538
+ "#7201a8"
539
+ ],
540
+ [
541
+ 0.3333333333333333,
542
+ "#9c179e"
543
+ ],
544
+ [
545
+ 0.4444444444444444,
546
+ "#bd3786"
547
+ ],
548
+ [
549
+ 0.5555555555555556,
550
+ "#d8576b"
551
+ ],
552
+ [
553
+ 0.6666666666666666,
554
+ "#ed7953"
555
+ ],
556
+ [
557
+ 0.7777777777777778,
558
+ "#fb9f3a"
559
+ ],
560
+ [
561
+ 0.8888888888888888,
562
+ "#fdca26"
563
+ ],
564
+ [
565
+ 1,
566
+ "#f0f921"
567
+ ]
568
+ ],
569
+ "type": "surface"
570
+ }
571
+ ],
572
+ "table": [
573
+ {
574
+ "cells": {
575
+ "fill": {
576
+ "color": "#EBF0F8"
577
+ },
578
+ "line": {
579
+ "color": "white"
580
+ }
581
+ },
582
+ "header": {
583
+ "fill": {
584
+ "color": "#C8D4E3"
585
+ },
586
+ "line": {
587
+ "color": "white"
588
+ }
589
+ },
590
+ "type": "table"
591
+ }
592
+ ]
593
+ },
594
+ "layout": {
595
+ "annotationdefaults": {
596
+ "arrowcolor": "#2a3f5f",
597
+ "arrowhead": 0,
598
+ "arrowwidth": 1
599
+ },
600
+ "autotypenumbers": "strict",
601
+ "coloraxis": {
602
+ "colorbar": {
603
+ "outlinewidth": 0,
604
+ "ticks": ""
605
+ }
606
+ },
607
+ "colorscale": {
608
+ "diverging": [
609
+ [
610
+ 0,
611
+ "#8e0152"
612
+ ],
613
+ [
614
+ 0.1,
615
+ "#c51b7d"
616
+ ],
617
+ [
618
+ 0.2,
619
+ "#de77ae"
620
+ ],
621
+ [
622
+ 0.3,
623
+ "#f1b6da"
624
+ ],
625
+ [
626
+ 0.4,
627
+ "#fde0ef"
628
+ ],
629
+ [
630
+ 0.5,
631
+ "#f7f7f7"
632
+ ],
633
+ [
634
+ 0.6,
635
+ "#e6f5d0"
636
+ ],
637
+ [
638
+ 0.7,
639
+ "#b8e186"
640
+ ],
641
+ [
642
+ 0.8,
643
+ "#7fbc41"
644
+ ],
645
+ [
646
+ 0.9,
647
+ "#4d9221"
648
+ ],
649
+ [
650
+ 1,
651
+ "#276419"
652
+ ]
653
+ ],
654
+ "sequential": [
655
+ [
656
+ 0,
657
+ "#0d0887"
658
+ ],
659
+ [
660
+ 0.1111111111111111,
661
+ "#46039f"
662
+ ],
663
+ [
664
+ 0.2222222222222222,
665
+ "#7201a8"
666
+ ],
667
+ [
668
+ 0.3333333333333333,
669
+ "#9c179e"
670
+ ],
671
+ [
672
+ 0.4444444444444444,
673
+ "#bd3786"
674
+ ],
675
+ [
676
+ 0.5555555555555556,
677
+ "#d8576b"
678
+ ],
679
+ [
680
+ 0.6666666666666666,
681
+ "#ed7953"
682
+ ],
683
+ [
684
+ 0.7777777777777778,
685
+ "#fb9f3a"
686
+ ],
687
+ [
688
+ 0.8888888888888888,
689
+ "#fdca26"
690
+ ],
691
+ [
692
+ 1,
693
+ "#f0f921"
694
+ ]
695
+ ],
696
+ "sequentialminus": [
697
+ [
698
+ 0,
699
+ "#0d0887"
700
+ ],
701
+ [
702
+ 0.1111111111111111,
703
+ "#46039f"
704
+ ],
705
+ [
706
+ 0.2222222222222222,
707
+ "#7201a8"
708
+ ],
709
+ [
710
+ 0.3333333333333333,
711
+ "#9c179e"
712
+ ],
713
+ [
714
+ 0.4444444444444444,
715
+ "#bd3786"
716
+ ],
717
+ [
718
+ 0.5555555555555556,
719
+ "#d8576b"
720
+ ],
721
+ [
722
+ 0.6666666666666666,
723
+ "#ed7953"
724
+ ],
725
+ [
726
+ 0.7777777777777778,
727
+ "#fb9f3a"
728
+ ],
729
+ [
730
+ 0.8888888888888888,
731
+ "#fdca26"
732
+ ],
733
+ [
734
+ 1,
735
+ "#f0f921"
736
+ ]
737
+ ]
738
+ },
739
+ "colorway": [
740
+ "#636efa",
741
+ "#EF553B",
742
+ "#00cc96",
743
+ "#ab63fa",
744
+ "#FFA15A",
745
+ "#19d3f3",
746
+ "#FF6692",
747
+ "#B6E880",
748
+ "#FF97FF",
749
+ "#FECB52"
750
+ ],
751
+ "font": {
752
+ "color": "#2a3f5f"
753
+ },
754
+ "geo": {
755
+ "bgcolor": "white",
756
+ "lakecolor": "white",
757
+ "landcolor": "#E5ECF6",
758
+ "showlakes": true,
759
+ "showland": true,
760
+ "subunitcolor": "white"
761
+ },
762
+ "hoverlabel": {
763
+ "align": "left"
764
+ },
765
+ "hovermode": "closest",
766
+ "mapbox": {
767
+ "style": "light"
768
+ },
769
+ "paper_bgcolor": "white",
770
+ "plot_bgcolor": "#E5ECF6",
771
+ "polar": {
772
+ "angularaxis": {
773
+ "gridcolor": "white",
774
+ "linecolor": "white",
775
+ "ticks": ""
776
+ },
777
+ "bgcolor": "#E5ECF6",
778
+ "radialaxis": {
779
+ "gridcolor": "white",
780
+ "linecolor": "white",
781
+ "ticks": ""
782
+ }
783
+ },
784
+ "scene": {
785
+ "xaxis": {
786
+ "backgroundcolor": "#E5ECF6",
787
+ "gridcolor": "white",
788
+ "gridwidth": 2,
789
+ "linecolor": "white",
790
+ "showbackground": true,
791
+ "ticks": "",
792
+ "zerolinecolor": "white"
793
+ },
794
+ "yaxis": {
795
+ "backgroundcolor": "#E5ECF6",
796
+ "gridcolor": "white",
797
+ "gridwidth": 2,
798
+ "linecolor": "white",
799
+ "showbackground": true,
800
+ "ticks": "",
801
+ "zerolinecolor": "white"
802
+ },
803
+ "zaxis": {
804
+ "backgroundcolor": "#E5ECF6",
805
+ "gridcolor": "white",
806
+ "gridwidth": 2,
807
+ "linecolor": "white",
808
+ "showbackground": true,
809
+ "ticks": "",
810
+ "zerolinecolor": "white"
811
+ }
812
+ },
813
+ "shapedefaults": {
814
+ "line": {
815
+ "color": "#2a3f5f"
816
+ }
817
+ },
818
+ "ternary": {
819
+ "aaxis": {
820
+ "gridcolor": "white",
821
+ "linecolor": "white",
822
+ "ticks": ""
823
+ },
824
+ "baxis": {
825
+ "gridcolor": "white",
826
+ "linecolor": "white",
827
+ "ticks": ""
828
+ },
829
+ "bgcolor": "#E5ECF6",
830
+ "caxis": {
831
+ "gridcolor": "white",
832
+ "linecolor": "white",
833
+ "ticks": ""
834
+ }
835
+ },
836
+ "title": {
837
+ "x": 0.05
838
+ },
839
+ "xaxis": {
840
+ "automargin": true,
841
+ "gridcolor": "white",
842
+ "linecolor": "white",
843
+ "ticks": "",
844
+ "title": {
845
+ "standoff": 15
846
+ },
847
+ "zerolinecolor": "white",
848
+ "zerolinewidth": 2
849
+ },
850
+ "yaxis": {
851
+ "automargin": true,
852
+ "gridcolor": "white",
853
+ "linecolor": "white",
854
+ "ticks": "",
855
+ "title": {
856
+ "standoff": 15
857
+ },
858
+ "zerolinecolor": "white",
859
+ "zerolinewidth": 2
860
+ }
861
+ }
862
+ },
863
+ "title": {
864
+ "text": "Prediction error from each model"
865
+ },
866
+ "width": 1000,
867
+ "xaxis": {
868
+ "tickangle": -45
869
+ },
870
+ "yaxis": {
871
+ "range": [
872
+ 0,
873
+ 110.72
874
+ ],
875
+ "title": {
876
+ "text": "Error"
877
+ }
878
+ }
879
+ }
880
+ }
881
+ },
882
+ "metadata": {},
883
+ "output_type": "display_data"
884
+ }
885
+ ],
886
+ "source": [
887
+ "import plotly.graph_objects as go\n",
888
+ "\n",
889
+ "results = [\n",
890
+ " (\"Constant\", \"gray\", 106.18),\n",
891
+ " (\"Linear Regression\", \"gray\", 101.56),\n",
892
+ " (\"NLP + LR\", \"gray\", 76.81),\n",
893
+ " (\"Random Forest\", \"gray\", 72.28),\n",
894
+ " (\"XGBoost\", \"gray\", 68.23),\n",
895
+ " (\"Human (Ed)\", \"black\", 87.62),\n",
896
+ " (\"Neural Network\", \"orange\", 63.97),\n",
897
+ " (\"GPT 4.1 Nano\", \"slateblue\", 62.51),\n",
898
+ " (\"Grok 4.1 Fast\", \"slateblue\", 57.62),\n",
899
+ " (\"Gemini 3 Pro\", \"slateblue\", 50.54),\n",
900
+ " (\"Claude 4.5 Sonnet\", \"slateblue\", 47.10),\n",
901
+ " (\"GPT 5.1\", \"slateblue\", 44.74),\n",
902
+ " (\"GPT 4.1 Nano (Fine-tuned)\", \"skyblue\", 75.91),\n",
903
+ " (\"Deep Neural Network\", \"orange\", 46.49),\n",
904
+ " (\"Base Llama 3.2 4 bit\", \"darkred\", 110.72),\n",
905
+ " (\"Fine-tuned Lite\", \"red\", 65.40),\n",
906
+ " (\"Fine-tuned Full\", \"red\", 39.85),\n",
907
+ " (\"GPT 5.1 RAG\", \"green\", 30.19),\n",
908
+ " (\"Ensemble\", \"purple\", 29.9)\n",
909
+ "]\n",
910
+ "\n",
911
+ "labels, colors, values = zip(*results)\n",
912
+ "\n",
913
+ "fig = go.Figure(go.Bar(x=labels, y=values, marker_color=colors))\n",
914
+ "\n",
915
+ "fig.update_layout(\n",
916
+ " title=\"Prediction error from each model\",\n",
917
+ " yaxis=dict(range=[0, max(values)], title=\"Error\"),\n",
918
+ " xaxis=dict(tickangle=-45),\n",
919
+ " width=1000,\n",
920
+ " height=800\n",
921
+ ")\n",
922
+ "\n",
923
+ "fig.show()"
924
+ ]
925
+ },
926
+ {
927
+ "cell_type": "code",
928
+ "execution_count": null,
929
+ "id": "bfac7653",
930
+ "metadata": {},
931
+ "outputs": [],
932
+ "source": []
933
+ },
934
+ {
935
+ "cell_type": "code",
936
+ "execution_count": null,
937
+ "id": "de22d815",
938
+ "metadata": {},
939
+ "outputs": [],
940
+ "source": []
941
+ }
942
+ ],
943
+ "metadata": {
944
+ "kernelspec": {
945
+ "display_name": ".venv",
946
+ "language": "python",
947
+ "name": "python3"
948
+ },
949
+ "language_info": {
950
+ "codemirror_mode": {
951
+ "name": "ipython",
952
+ "version": 3
953
+ },
954
+ "file_extension": ".py",
955
+ "mimetype": "text/x-python",
956
+ "name": "python",
957
+ "nbconvert_exporter": "python",
958
+ "pygments_lexer": "ipython3",
959
+ "version": "3.12.12"
960
+ }
961
+ },
962
+ "nbformat": 4,
963
+ "nbformat_minor": 5
964
+ }