Skip to content

Commit 7217d94

Browse files
committed
add mlflow tracker example
1 parent 07cb38e commit 7217d94

File tree

3 files changed

+4
-9
lines changed

3 files changed

+4
-9
lines changed

examples/sparse_lr/example_2.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,10 @@
1-
from hyperparameter.hp import param_scope, Tracker
1+
from hyperparameter import param_scope, Tracker
22
from model import sparse_lr_plot
3-
import numpy as np
43

54
from sklearn import datasets
65
from sklearn.preprocessing import StandardScaler
7-
import matplotlib.pyplot as plt
86

97
X, y = datasets.load_digits(return_X_y=True)
10-
118
X = StandardScaler().fit_transform(X)
129

1310
# classify small against large digits
@@ -20,7 +17,6 @@ def run(args):
2017
sparse_lr_plot(X, y)
2118

2219
if __name__ == '__main__':
23-
# create cmd line arguments parser
2420
import argparse
2521
parser = argparse.ArgumentParser('example')
2622
parser.add_argument('-D', '--define', nargs='*', default=[], action="extend",

examples/sparse_lr/model.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44

55
from hyperparameter import auto_param, param_scope, set_tracker
66

7-
set_tracker(lambda x: print(x))
8-
97
MyLogisticRegression = auto_param(LogisticRegression)
108

119

hyperparameter/hp.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -375,10 +375,11 @@ def wrapper(*arg, **kws):
375375
with param_scope() as hp:
376376
local_params = {}
377377
for k, v in predef_kws.items():
378-
local_params[v] = predef_val[v]
379-
if hp.get(v) is not None:
378+
if hp.get(v) is not None and k not in kws:
380379
kws[k] = hp.get(v)
381380
local_params[v] = hp.get(v)
381+
else:
382+
local_params[v] = predef_val[v]
382383
if Tracker.callback is not None:
383384
Tracker.callback(local_params)
384385
return func(*arg, **kws)

0 commit comments

Comments
 (0)