search

IIIS-Li-Group / OpenFE

OpenFE: automated feature generation with expert-level performance
MIT License
764 stars 98 forks source link

support GPU？ #37

Open 280185386 opened 10 months ago

280185386 commented 10 months ago

does openFE support GPU

bencoldham commented 3 months ago

does openFE support GPU

Not that I know of. I tried switching to gpu by updating the lgbm parameters in openfe.py to "device":"cuda" and whilst it worked, I noticed it greatly increased runtime.

I guess with the parameters of stage-1 being:

 params = {"n_estimators": 100, "importance_type": "gain", "num_leaves": 16,
                          "seed": 1, "deterministic": True, "n_jobs": 1, "verbose": 1 if self.verbose else -1 }

the overhead of GPU usage might outweigh the increased training speed in the current implementation.