|
@@ -6,6 +6,7 @@ from sklearn.ensemble import RandomForestRegressor
|
6
|
6
|
from sklearn.datasets import load_wine
|
7
|
7
|
from sklearn.model_selection import train_test_split
|
8
|
8
|
import numpy as np
|
|
9
|
+from util.config import config
|
9
|
10
|
from sklearn.tree import DecisionTreeRegressor
|
10
|
11
|
from sklearn import metrics
|
11
|
12
|
|
|
@@ -39,8 +40,8 @@ def read_data(path):
|
39
|
40
|
|
40
|
41
|
|
41
|
42
|
def demo():
|
42
|
|
- X_train, y_train = read_data("../bbztx/train_data")
|
43
|
|
- X_test, y_test = read_data("../bbztx/test_data")
|
|
43
|
+ X_train, y_train = read_data(config.get('application', 'train_data_path'))
|
|
44
|
+ X_test, y_test = read_data(config.get('application', 'train_data_path'))
|
44
|
45
|
Xtrain, Xtest, Ytrain, Ytest = train_test_split(X_train, y_train, test_size=0.3)
|
45
|
46
|
rfc = RandomForestRegressor(random_state=0, n_estimators=10, max_depth=10)
|
46
|
47
|
rfc = rfc.fit(Xtrain, Ytrain)
|