mirror of
https://github.com/vale981/ray
synced 2025-03-10 21:36:39 -04:00
14 lines
382 B
Python
14 lines
382 B
Python
|
import ray
|
||
|
|
|
||
|
|
from ray.train.lightgbm import LightGBMTrainer
|
||
|
|
from ray.air.config import ScalingConfig
|
||
|
|
|
||
|
|
train_dataset = ray.data.from_items([{"x": x, "y": x + 1} for x in range(32)])
|
||
|
|
trainer = LightGBMTrainer(
|
||
|
|
label_column="y",
|
||
|
|
params={"objective": "regression"},
|
||
|
|
scaling_config=ScalingConfig(num_workers=3),
|
||
|
|
datasets={"train": train_dataset},
|
||
|
|
)
|
||
|
|
result = trainer.fit()
|