Commit 629b0fcc authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Some tweaks

parent ecb7160a

examples/low_data/tox_attn_one_fold.py

+2 −2
Original line number Diff line number Diff line
@@ -15,8 +15,8 @@ from datasets import load_tox21_convmol 
K = 4 

# Depth of attention module

max_depth = 4

# 10 positive/negative ligands

n_pos = 10

# number positive/negative ligands

n_pos = 5 

n_neg = 10

# Set batch sizes for network

test_batch_size = 100

examples/low_data/tox_rf_one_fold.py

+2 −2
Original line number Diff line number Diff line
@@ -21,8 +21,8 @@ model_dir = tempfile.mkdtemp() 


# 4-fold splits

K = 4

# 10 positive/negative ligands

n_pos = 10

# num positive/negative ligands

n_pos = 5

n_neg = 10

# 10 trials on test-set

n_trials = 10

examples/low_data/tox_siamese_one_fold.py

+2 −2
Original line number Diff line number Diff line
@@ -13,8 +13,8 @@ from datasets import load_tox21_convmol 


# Number of folds for split 

K = 4

# 10 positive/negative ligands

n_pos = 10

# num positive/negative ligands

n_pos = 5

n_neg = 10

# Set batch sizes for network

test_batch_size = 100

CRA Git | Maintained and supported by SUSTech CRA and CCSE