Rename public-facing CodonGPT strings to CodonTranslator
Browse files- sampling.py +3 -3
sampling.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
#!/usr/bin/env python
|
| 2 |
"""
|
| 3 |
-
Sampling script for generating codon sequences from trained
|
| 4 |
Inputs are prepared exactly like training:
|
| 5 |
- Species conditioning via SpeciesEmbeddingStore (fixed-size [B,Ds] or variable-length [B,Ls,Ds])
|
| 6 |
- Protein conditioning via raw AA strings (ESM-C tokenization happens inside the model)
|
|
@@ -22,11 +22,11 @@ logging.basicConfig(
|
|
| 22 |
datefmt="%m/%d/%Y %H:%M:%S",
|
| 23 |
level=logging.INFO,
|
| 24 |
)
|
| 25 |
-
logger = logging.getLogger("
|
| 26 |
|
| 27 |
|
| 28 |
def parse_args():
|
| 29 |
-
p = argparse.ArgumentParser(description="Sample codon sequences from
|
| 30 |
|
| 31 |
# Model
|
| 32 |
p.add_argument("--model_path", "--model_dir", dest="model_path", type=str, required=True,
|
|
|
|
| 1 |
#!/usr/bin/env python
|
| 2 |
"""
|
| 3 |
+
Sampling script for generating codon sequences from trained CodonTranslator models.
|
| 4 |
Inputs are prepared exactly like training:
|
| 5 |
- Species conditioning via SpeciesEmbeddingStore (fixed-size [B,Ds] or variable-length [B,Ls,Ds])
|
| 6 |
- Protein conditioning via raw AA strings (ESM-C tokenization happens inside the model)
|
|
|
|
| 22 |
datefmt="%m/%d/%Y %H:%M:%S",
|
| 23 |
level=logging.INFO,
|
| 24 |
)
|
| 25 |
+
logger = logging.getLogger("codontranslator.sample")
|
| 26 |
|
| 27 |
|
| 28 |
def parse_args():
|
| 29 |
+
p = argparse.ArgumentParser(description="Sample codon sequences from a CodonTranslator model")
|
| 30 |
|
| 31 |
# Model
|
| 32 |
p.add_argument("--model_path", "--model_dir", dest="model_path", type=str, required=True,
|