TODO: test ensemble with different data sets