Commit
·
e1370eb
1
Parent(s):
8aec0bb
Fixed bug on shuffle_embedding_prob in PROTAC_Dataset class
Browse files
protac_degradation_predictor/protac_dataset.py
CHANGED
|
@@ -89,8 +89,8 @@ class PROTAC_Dataset(Dataset):
|
|
| 89 |
if self.use_smote:
|
| 90 |
self.apply_smote()
|
| 91 |
|
|
|
|
| 92 |
if shuffle_embedding_prob > 0.0:
|
| 93 |
-
self.shuffle_embedding_prob = shuffle_embedding_prob
|
| 94 |
# Set random seed
|
| 95 |
random.seed(42)
|
| 96 |
if self.protein_emb_dim != self.cell_emb_dim:
|
|
|
|
| 89 |
if self.use_smote:
|
| 90 |
self.apply_smote()
|
| 91 |
|
| 92 |
+
self.shuffle_embedding_prob = shuffle_embedding_prob
|
| 93 |
if shuffle_embedding_prob > 0.0:
|
|
|
|
| 94 |
# Set random seed
|
| 95 |
random.seed(42)
|
| 96 |
if self.protein_emb_dim != self.cell_emb_dim:
|