felixm/ML4T
1
0
Fork 0
Code Releases Activity

Resolve split_value issue in DTLearner and pass all tests.

Browse Source
This commit is contained in:
Felix Martin
2020-09-25 10:27:15 -04:00
parent 7007bc7514
commit bd19b4fb18
3 changed files with 21 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
assess_learners/testlearner.py
View File

@@ -48,12 +48,6 @@ if __name__=="__main__":
trainY = data[:train_rows,-1]
testX = data[train_rows:,0:-1]
testY = data[train_rows:,-1]
# trainX = data[:, 0:-1]
# trainY = data[:, -1]
# testX = data[:, 0:-1]
# testY = data[:, -1]
print(f"{testX.shape}")
print(f"{testY.shape}")
@@ -85,8 +79,7 @@ if __name__=="__main__":
# test_learner(lrl.LinRegLearner)
test_learner(dtl.DTLearner, leaf_size=1)
test_learner(rtl.RTLearner)
test_learner(rtl.RTLearner, leaf_size=5)
# test_learner(bgl.BagLearner, learner=dtl.DTLearner, bags=20)
# learner = isl.InsaneLearner()
# test_learner(rtl.RTLearner, leaf_size=6)
test_learner(bgl.BagLearner, learner=dtl.DTLearner, bags=20, kwargs = {'leaf_size': 5})
test_learner(isl.InsaneLearner)