|
@@ -41,7 +41,7 @@ def read_data(path):
|
41
|
41
|
|
42
|
42
|
def demo():
|
43
|
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
|
+ X_test, y_test = read_data(config.get('application', 'test_data_path'))
|
45
|
45
|
Xtrain, Xtest, Ytrain, Ytest = train_test_split(X_train, y_train, test_size=0.3)
|
46
|
46
|
rfc = RandomForestRegressor(random_state=0, n_estimators=10, max_depth=10)
|
47
|
47
|
rfc = rfc.fit(Xtrain, Ytrain)
|