import sys, csv, xalglib, time

train_file = sys.argv[1]
test_file = sys.argv[2]

train = []
reader = csv.reader( open( train_file ))
for line in reader:
	train.append( line[1:] )
	
test = []
test_y = []
test_x = []
reader = csv.reader( open( test_file ))
for line in reader:
	test.append( line[1:] )
	test_y.append( line[1] )
	test_x.append( line[2:] )


print "data loaded."
tic = time.time()
	
info, df, report = xalglib.dfbuildrandomdecisionforest( train, len( train ), 1, 1, 50, 0.5 )

print "%s seconds elapsed.\n" % ( time.time() - tic )

for i in report.__dict__:
	print "%s: %s" % ( i, report.__dict__[i] )	

'''
print	
print xalglib.dfavgerror( df, test, len( test ))
print xalglib.dfavgrelerror( df, test, len( test ))
print xalglib.dfrmserror( df, test, len( test ))
'''

for x in test_x:
	p = xalglib.dfprocessi( df, x )
	print p