Simple application of sklearn random forest algorithm