-
Notifications
You must be signed in to change notification settings - Fork 88
Open
Description
I am new to thundergbm, and just trying to get a simple Random Forest classifier going. But the classifier classifies every single sample to be 1. Not one single case out of 188244 samples is classified as 0. No other classifier behaves like this. I also tried different number of trees, depth etc. But it still classies everything to 1. Is there something wrong with the following code?
from thundergbm import TGBMClassifier
clf = TGBMClassifier(depth=6, n_trees = 1, n_parallel_trees=100, bagging=1)
clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)
#y_pred classifies everything in the test set (X_test) to one.
Metadata
Metadata
Assignees
Labels
No labels