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))