Skip to content

Commit

Permalink
rename MLAP files
Browse files Browse the repository at this point in the history
  • Loading branch information
jonasrothfuss committed Jun 4, 2020
1 parent a011856 commit 7098ca2
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion experiments/compuational_comparison.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
mean_nn_layers=NN_LAYERS, svi_batch_size=5,
kernel_nn_layers=NN_LAYERS, cov_type='diag', normalize_data=True)

from meta_learn.GPR_meta_pac import GPRegressionMetaLearnedPAC
from meta_learn.GPR_meta_mlap import GPRegressionMetaLearnedPAC

model_mlap = GPRegressionMetaLearnedPAC(meta_train_data, num_iter_fit=1000,
svi_batch_size=5, covar_module='NN', mean_module='NN', mean_nn_layers=NN_LAYERS,
Expand Down
2 changes: 1 addition & 1 deletion experiments/hyperparam_search/launch_hyperparam_sweeps.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

import experiments.hyperparam_search.meta_svgd_hyperparam as meta_svgd_hparam
import experiments.hyperparam_search.meta_vi_hyperparam as meta_vi_hparam
import experiments.hyperparam_search.meta_pac_hyperparam as meta_pac_hyperparam
import experiments.hyperparam_search.meta_mlap_hyperparam as meta_pac_hyperparam

from absl import flags
from absl import app
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ def train_reg(config, reporter):
data_train, data_valid, _ = provide_data(dataset=args.dataset, seed=SEED)

# 2) setup model
from meta_learn.GPR_meta_pac import GPRegressionMetaLearnedPAC
from meta_learn.GPR_meta_mlap import GPRegressionMetaLearnedPAC
torch.set_num_threads(N_THREADS_PER_RUN)

model = GPRegressionMetaLearnedPAC(data_train, **config)
Expand Down Expand Up @@ -74,7 +74,7 @@ def train_test(config):
data_train, _, data_test = provide_data(dataset=args.dataset, seed=SEED)

# 2) Fit model
from meta_learn.GPR_meta_pac import GPRegressionMetaLearnedPAC
from meta_learn.GPR_meta_mlap import GPRegressionMetaLearnedPAC
torch.set_num_threads(N_THREADS_PER_RUN)
with gpytorch.settings.max_cg_iterations(500):
model = GPRegressionMetaLearnedPAC(data_train, **config)
Expand Down
File renamed without changes.

0 comments on commit 7098ca2

Please sign in to comment.