*update* May be it doesn't have to be done at the tree creation level. It could be using loops and creating two different balltrees. Something like
tree1=BallTree(X,metric='metric1') #for x-z plane tree2=BallTree(X,metric='metric2') #for y-z plane And then calculate correlation functions in a loop to get tpcf(X,r1,r2) using tree1.two_point_correlation(X,r1) and tree2.two_point_correlation(X,r2)
_______________________________________________ scikit-learn mailing list scikit-learn@python.org https://mail.python.org/mailman/listinfo/scikit-learn