(task_type)
| 2290 | |
| 2291 | |
| 2292 | def test_ones_weight(task_type): |
| 2293 | pool = Pool(TRAIN_FILE, column_description=CD_FILE) |
| 2294 | weight = np.ones(pool.num_row()) |
| 2295 | pool.set_weight(weight) |
| 2296 | model = CatBoostClassifier(iterations=2, learning_rate=0.03, task_type=task_type, gpu_ram_part=TEST_GPU_RAM_PART, devices='0') |
| 2297 | model.fit(pool) |
| 2298 | output_model_path = test_output_path(OUTPUT_MODEL_PATH) |
| 2299 | model.save_model(output_model_path) |
| 2300 | return compare_canonical_models(output_model_path) |
| 2301 | |
| 2302 | |
| 2303 | def test_non_ones_weight(task_type): |
nothing calls this directly
no test coverage detected