pub_soft/ScanNet
3
0
Fork 0
mirror of https://github.com/jertubiana/ScanNet.git synced 2026-06-04 13:44:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Small bug fixes

Browse Source
This commit is contained in:
Jérôme Tubiana
2023-06-13 13:36:49 +03:00
parent 7407c1ae8c
commit 7685549c8e
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
preprocessing/pipelines.py
View File

@@ -199,6 +199,7 @@ class Pipeline():
print('Saving processed dataset...')
t = time.time()
env = {'inputs': inputs, 'outputs': outputs,'failed_samples':failed_samples}
os.makedirs(os.path.dirname(location_processed_dataset),exist_ok=True)
io_utils.save_pickle(
env, location_processed_dataset)
print('Processed dataset saved (t=%.f s)' % (time.time() - t))

2
train.py
View File

@@ -77,7 +77,7 @@ if __name__ == '__main__':
'''
check = False # Check = True to verify installation, =False for full training.
train = True # True to retrain, False to evaluate the model shown in paper.
use_evolutionary = True # True to use evolutionary information (requires hhblits and a sequence database), False otherwise.
use_evolutionary = False # True to use evolutionary information (requires hhblits and a sequence database), False otherwise.
Lmax_aa = 256 if check else 1024
'''
Maximum length of the protein sequences.

2
transfer_learning_train.py
View File

@@ -21,7 +21,7 @@ if __name__ == '__main__':
train = True # True to retrain, False to evaluate the model shown in paper.
transfer = True # If False, retrain from scratch.
freeze = False # If True, evaluate the binding site network without fine tuning.
use_evolutionary = True # True to use evolutionary information (requires hhblits and a sequence database), False otherwise.
use_evolutionary = False # True to use evolutionary information (requires hhblits and a sequence database), False otherwise.
Lmax_aa = 256 if check else 2120
'''
Lmax_aa is the maximum length of the protein sequences.