1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from __future__ import division, print_function
import sys
print("\nexecuting test file", __file__, file=sys.stderr)
exec(compile(open('set_paths.py', "rb").read(), 'set_paths.py', 'exec'))
import vigra
import numpy as np
gaus1=10*np.random.randn(1000).reshape((500,2))+np.array([20,20])
gaus1=gaus1.astype(np.float32)
gaus2=10*np.random.randn(1000).reshape((500,2))+np.array([20,20])
gaus2=gaus2.astype(np.float32)
label_gaus1=np.ones(500).astype(np.uint32)
label_gaus2=np.zeros(500).astype(np.uint32)
def test_rf_learn():
RF=vigra.learning.RandomForest(10)
fmat=np.vstack([gaus1,gaus2])
lmat=np.vstack([label_gaus1,label_gaus2]).reshape(-1,1)
RF.learnRF(fmat,lmat,0,1,100)
def ok_():
print(".", file=sys.stderr)
|
