@@ -1292,7 +1292,7 @@ def data_train_cell_activity(config, erase, best_model, device):
12921292 plt .xlabel (r'$a_{i0}$' , fontsize = 48 )
12931293 plt .ylabel (r'$a_{i1}$' , fontsize = 48 )
12941294
1295- stats_text = f'$\mu$={ model .a .mean ():.3f} \n $\sigma$={ model .a .std ():.3f} \n $\|grad\|$={ model .a .grad .norm ():.3f} '
1295+ stats_text = f'$\\ mu$={ model .a .mean ():.3f} \n $\\ sigma$={ model .a .std ():.3f} \n $\\ |grad\ \ |$={ model .a .grad .norm ():.3f} '
12961296 plt .text (0.02 , 0.98 , stats_text , transform = ax .transAxes ,
12971297 fontsize = 24 , verticalalignment = 'top' )
12981298
@@ -1323,7 +1323,7 @@ def data_train_cell_activity(config, erase, best_model, device):
13231323 plt .scatter (to_numpy (model .edges_embedding [:, 0 ]), to_numpy (model .edges_embedding [:, 1 ]), s = 10 , color = 'r' , alpha = 0.05 , edgecolor = 'none' )
13241324 plt .xlabel (r'$e_{ij,0}$' , fontsize = 48 )
13251325 plt .ylabel (r'$e_{ij,1}$' , fontsize = 48 )
1326- stats_text = f'$\mu$={ model .edges_embedding .mean ():.3f} \n $\sigma$={ model .edges_embedding .std ():.3f} \n $\|grad\|$={ model .edges_embedding .grad .norm ():.3f} '
1326+ stats_text = f'$\\ mu$={ model .edges_embedding .mean ():.3f} \n $\\ sigma$={ model .edges_embedding .std ():.3f} \n $\\ |grad\ \ |$={ model .edges_embedding .grad .norm ():.3f} '
13271327 plt .text (0.02 , 0.98 , stats_text , transform = ax .transAxes ,
13281328 fontsize = 24 , verticalalignment = 'top' )
13291329 plt .tight_layout ()
0 commit comments