We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 624eec7 commit 5df52f3Copy full SHA for 5df52f3
tests/modules/scoring/test_catboost.py
@@ -55,7 +55,7 @@ def test_catboost_prediction(dataset):
55
data_handler = DataHandler(dataset)
56
57
scorer = CatBoostScorer(
58
- classification_model_config="prajjwal1/bert-tiny",
+ embedder_config="prajjwal1/bert-tiny",
59
iterations=50,
60
learning_rate=0.05,
61
depth=6,
@@ -95,7 +95,7 @@ def test_catboost_prediction_multilabel(dataset):
95
data_handler = DataHandler(dataset.to_multilabel())
96
97
98
99
100
101
0 commit comments