Closed johannes-kk closed 4 years ago
@johannes-kk see PR #89 Taking X and y separately is less straightforward because I don't think we have an efficient way of dropping the y column from an existing DataFrame, but feel free to give it a try.
No need. Keep it simple to cut down on time. What we have already works.
Currently the training
DataFrame
is passed to theDecisionTree
orRandomForest
constructor. Most APIs pass train and test data to.fit(train_x, train_y)
, not to the constructor. Make ours consistent with that.