just tweaking. still needs fix from https://0xdata.atlassian.net/browse/HEX-549
rf train and test on different datasets