Update plot_tree_regression.py

Import statements have been moved to the beginning of the file.