Unverified Commit 112b71f8 authored by Aolong's avatar Aolong Committed by GitHub
Browse files

change a parameter name

`logdir` is meaningless. Change to `model_dir` makes this parameter useful.
parent 25e38a21
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -34,7 +34,7 @@ n_features = train_dataset.X.shape[1]
model = dc.models.MultitaskRegressor(
    len(pdbbind_tasks),
    n_features,
    logdir=model_dir,
    model_dir=model_dir,
    dropouts=[.25],
    learning_rate=0.0003,
    weight_init_stddevs=[.1],