ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix lightgbm binding keyword argument issue (#251)

Browse Source
This commit is contained in:
noklam 2020-11-24 19:27:33 +08:00 committed by GitHub
parent a3ad6a0e50
commit 6064080232
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
trains/binding/frameworks/lightgbm_bind.py
View File

@ -118,11 +118,12 @@ class PatchLIGHTgbmModelIO(PatchBaseModelIO):
except Exception: except Exception:
pass pass
return callback return callback
params, train_set = args
kwargs.setdefault("callbacks", []).append(trains_lightgbm_callback()) kwargs.setdefault("callbacks", []).append(trains_lightgbm_callback())
ret = original_fn(params, train_set, **kwargs) ret = original_fn(*args, **kwargs)
if not PatchLIGHTgbmModelIO.__main_task: if not PatchLIGHTgbmModelIO.__main_task:
return ret return ret
params = args[0] if args else kwargs['params']
for k, v in params.items(): for k, v in params.items():
if isinstance(v, set): if isinstance(v, set):
params[k] = list(v) params[k] = list(v)