@@ -352,6 +352,15 @@ def train_model_diversity(
352
352
pass
353
353
354
354
plt .figure (figsize = (12 , 6 ))
355
+
356
+ plt .rc ('font' , size = 16 ) # controls default text sizes
357
+ plt .rc ('axes' , titlesize = 16 ) # fontsize of the axes title
358
+ plt .rc ('axes' , labelsize = 16 ) # fontsize of the x and y labels
359
+ plt .rc ('xtick' , labelsize = 16 ) # fontsize of the tick labels
360
+ plt .rc ('ytick' , labelsize = 16 ) # fontsize of the tick labels
361
+ plt .rc ('legend' , fontsize = 16 ) # legend fontsize
362
+ plt .rc ('figure' , titlesize = 16 ) # fontsize of the figure title
363
+
355
364
plt .plot (
356
365
range (1 , len (train_losses ) + 1 ),
357
366
train_losses ,
@@ -451,6 +460,15 @@ def train_model_accuracy(
451
460
pass
452
461
453
462
plt .figure (figsize = (12 , 6 ))
463
+
464
+ plt .rc ('font' , size = 16 ) # controls default text sizes
465
+ plt .rc ('axes' , titlesize = 16 ) # fontsize of the axes title
466
+ plt .rc ('axes' , labelsize = 16 ) # fontsize of the x and y labels
467
+ plt .rc ('xtick' , labelsize = 16 ) # fontsize of the tick labels
468
+ plt .rc ('ytick' , labelsize = 16 ) # fontsize of the tick labels
469
+ plt .rc ('legend' , fontsize = 16 ) # legend fontsize
470
+ plt .rc ('figure' , titlesize = 16 ) # fontsize of the figure title
471
+
454
472
plt .plot (
455
473
range (1 , len (train_losses ) + 1 ),
456
474
train_losses ,
@@ -466,6 +484,7 @@ def train_model_accuracy(
466
484
plt .title ("Training and Validation Loss Over Epochs" )
467
485
plt .xlabel ("Epoch" )
468
486
plt .ylabel ("Loss" )
487
+ plt .ylim (0 , 0.002 )
469
488
plt .grid (True )
470
489
plt .legend ()
471
490
plt .show ()
0 commit comments