Skip to content

Commit

Permalink
more typing
Browse files Browse the repository at this point in the history
  • Loading branch information
Marvin84 committed Sep 21, 2023
1 parent 718808b commit ee176df
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions rasr/feature_scorer.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,13 +151,13 @@ def __init__(
:param io_map: mapping between internal rasr identifiers and the model related input/output. Default key values
are "features" and "output", and optionally "features-size", e.g.
io_map = {"features": "data", "output": "classes"}
:param float label_log_posterior_scale: scales for the log probability of a label e.g. 1.0 is recommended
:param float label_prior_scale: scale for the prior log probability of a label reasonable e.g. values in [0.1, 0.7] interval
:param str|tk.Path label_log_prior_file: xml file containing log prior probabilities e.g. estimated from the model via povey method
:param bool apply_log_on_output: whether to apply the log-function on the output, usefull if the model outputs softmax instead of log-softmax
:param bool negate_output: whether negate output (because the model outputs log softmax and not negative log softmax
:param int intra_op_threads: Onnxruntime session's number of parallel threads within each operator
:param int inter_op_threads: Onnxruntime session's number of parallel threads between operators used only for parallel execution mode
:param label_log_posterior_scale: scales for the log probability of a label e.g. 1.0 is recommended
:param label_prior_scale: scale for the prior log probability of a label reasonable e.g. values in [0.1, 0.7] interval
:param label_log_prior_file: xml file containing log prior probabilities e.g. estimated from the model via povey method
:param apply_log_on_output: whether to apply the log-function on the output, usefull if the model outputs softmax instead of log-softmax
:param negate_output: whether negate output (because the model outputs log softmax and not negative log softmax
:param intra_op_threads: Onnxruntime session's number of parallel threads within each operator
:param inter_op_threads: Onnxruntime session's number of parallel threads between operators used only for parallel execution mode
"""
super().__init__(*args, **kwargs)

Expand Down

0 comments on commit ee176df

Please sign in to comment.