diff --git a/train.py b/train.py index f4ff478f6..32d0c2a53 100644 --- a/train.py +++ b/train.py @@ -2,7 +2,7 @@ from sklearn.model_selection import train_test_split from sklearn.ensemble import RandomForestRegressor import matplotlib.pyplot as plt -import seaborn as sns +import seaborn as sns import numpy as np # Set random seed seed = 42