在算法伪代码环境中编写多行方程

在算法伪代码环境中编写多行方程

我有以下代码,它可以生成图像中的算法。

\documentclass{article}
\usepackage{algorithm}
\usepackage{algorithmicx}
\usepackage{algpseudocode}
\begin{document}
    \begin{algorithm}[h]
        \caption{Q-learning}
        \begin{algorithmic}[1]
            \Require Set of good nodes $V^g$, $score(u) \; \forall u \in V^g, \mbox{ number of episodes } E, \mbox{ budget size } b$, $\epsilon$
            \Ensure Learned parameters $\Theta_Q$
            \State Initialise replay buffer $M = \emptyset$
            \For{Episode $e \leftarrow 1 \mbox{ to } E$}
                \State $S_t \leftarrow \emptyset,\; C_t \leftarrow V^g$
                \For{Step $t \leftarrow 1 \mbox{ to } b$}
                    \State 
                    \begin{equation}
                        v_t \leftarrow 
                        \begin{cases}
                            \mbox{A node chosen uniformly at random from } C_t & \mbox{w.p. } \epsilon \\
                            \argmax_{v \in C_t} Q(S_t, v) & \mbox{w.p. } 1-\epsilon
                        \end{cases}
                    \end{equation}
                \EndFor
            \EndFor
        \end{algorithmic}
    \end{algorithm}
\end{document}

问题图片

我希望第 5 行不要位于页面中间,而应位于 for 循环下方(如果是普通文本的话)。我可以使用 formula/cases 环境来执行此操作吗?还是需要以其他方式创建它?

抱歉,如果我的公式等的导入信息缺失,因为它可能在我的序言中

相关内容