Decision-Tree-Simulation-in-Python

I try to run a simple Decision Tree simulation of Movie collectiond data by finding out how each independent variables affect budget. Using python, we can visualize the result of the decision.

1. import related libraries

2. Extracting the data Movie_regression.csv into pandas dataframe

3. Treat the missing value by replacing with mean.

4. Creating dummy variable, for variable with string value

5. Define X variable and y variable

6. Use test train split with test size: 0.2

7. import decision tree regressor, and define prediction of y

8. Find mean squared error and r2 score, define dot data

9. Display regression tree graph