Skip to content

Commit 74cc2c9

Browse files
committed
multiregression: Minor fixes
1 parent 01717c6 commit 74cc2c9

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

examples/multiregression_trees/multiregression_run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,10 +48,10 @@ def predict(self, features : array.array, outputs : array.array):
4848
def main():
4949

5050
# FIXME: read paths from sys.argv
51-
model = MultiRegressor()
51+
model = MultiRegressor(max_nodes=10000)
5252
model.load('models')
5353

54-
outputs = array.array('f', [0.0 for _ in range(model.models)])
54+
outputs = array.array('f', [0.0 for _ in range(len(model.models))])
5555

5656
import npyfile
5757
(n_samples, n_features), data = npyfile.load('data.npy')

0 commit comments

Comments
 (0)