restructure umami to remove tf dependencies for plotting
This MR moves some functions around to avoid tensor-flow dependencies for plotting related tasks.
It also modifies the output of the plot_input_variables.py script to remove the hard-coded prefix.