examples.test_scripts
Module Contents
Classes
Functions
|
|
|
|
|
- class examples.test_scripts.BoxArgs
Bases:
CommonArgs- delta :float = 0.25
- gamma_scheduler :float = 0.5
- lr_F :float = 0.01
- max_concentration :float = 5.1
- min_concentration :float = 0.1
- n_components :int = 2
- n_components_s0 :int = 4
- scheduler_milestone :int = 2500
- class examples.test_scripts.CommonArgs
- batch_size :int = 16
- loss :str = TB
- lr :float = 0.001
- lr_Z :float = 0.1
- n_trajectories :int = 32000
- no_cuda :bool = True
- replay_buffer_size :int = 0
- seed :int = 1
- subTB_lambda :float = 0.9
- subTB_weighting :str = geometric_within
- tabular :bool = False
- tied :bool = False
- uniform_pb :bool = False
- validation_interval :int = 100
- validation_samples :int = 200000
- wandb_project :str =
- class examples.test_scripts.DiscreteEBMArgs
Bases:
CommonArgs- alpha :float = 1.0
- ndim :int = 4
- class examples.test_scripts.HypergridArgs
Bases:
CommonArgs- R0 :float = 0.1
- R1 :float = 0.5
- R2 :float = 2.0
- height :int = 8
- ndim :int = 2
- examples.test_scripts.test_box(delta, loss)
- Parameters
delta (float) –
loss (str) –
- examples.test_scripts.test_discreteebm(ndim, alpha)
- Parameters
ndim (int) –
alpha (float) –
- examples.test_scripts.test_hypergrid(ndim, height)
- Parameters
ndim (int) –
height (int) –