Todo: more efficient graph presentation without using networkx more training modes function to display the weights of the features graph output with weight as edge label