Add random use case to docstring of predict_dna_sequence.

Adibvafa · Sep 20, 2024 · c655e22 · c655e22
1 parent 59e4c5c
commit c655e22
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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))