From c655e22c3c3cf710a71f81080eeff02a5d853824 Mon Sep 17 00:00:00 2001 From: Adibvafa Date: Fri, 20 Sep 2024 11:45:14 -0400 Subject: [PATCH] Add random use case to docstring of predict_dna_sequence. --- CodonTransformer/CodonPrediction.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/CodonTransformer/CodonPrediction.py b/CodonTransformer/CodonPrediction.py index 74922e7..8c5a4f6 100644 --- a/CodonTransformer/CodonPrediction.py +++ b/CodonTransformer/CodonPrediction.py @@ -123,7 +123,7 @@ def predict_dna_sequence( ... deterministic=True ... ) >>> - >>> # Predict DNA sequence with medium randomness + >>> # Predict DNA sequence with low randomness >>> output_random = predict_dna_sequence( ... protein=protein, ... organism=organism, @@ -132,7 +132,7 @@ def predict_dna_sequence( ... model=model, ... attention_type="original_full", ... deterministic=False, - ... temperature=1.0 + ... temperature=0.2 ... ) >>> >>> print(format_model_output(output))