[已解决]np.numpy.choice

万梓柠

batch = np.random.choice(mem_len, self.batch_size, replace=True)
求助各路大神，这里的np.random.choice的用法是什么，还有replace=True是什么意思

最佳答案

月排行榜 / 总排行榜

isdkz

2023-4-21 11:27:43

`np.random.choice`是NumPy库中的一个函数，用于从给定的一维数组或整数范围中随机抽取指定数量的样本。`np.random.choice`函数的基本语法如下：

numpy.random.choice(a, size=None, replace=True, p=None)

参数说明：

- `a`：一个一维数组或一个整数。如果是一个整数，那么它表示从`range(a)`中抽取样本。
- `size`：指定抽取样本的数量。可以是一个整数或者表示数组形状的元组。
- `replace`：一个布尔值，表示是否允许重复抽取。默认为`True`，即允许重复抽取。如果设置为`False`，那么抽取的样本将不重复。
- `p`：一个一维数组，表示抽取每个元素的概率。数组长度应与`a`相同。如果未指定，将使用均匀分布进行抽样。

在您的代码中：

batch = np.random.choice(mem_len, self.batch_size, replace=True)

`mem_len`可能是一个整数，表示从`range(mem_len)`中抽取样本；`self.batch_size`表示要抽取的样本数量；`replace=True`表示允许在抽取过程中重复选取相同的元素。所以，这行代码的作用是从`range(mem_len)`中随机抽取`self.batch_size`个样本，允许重复抽取。

跳转到最佳答案楼层

isdkz

`np.random.choice`是NumPy库中的一个函数，用于从给定的一维数组或整数范围中随机抽取指定数量的样本。`np.random.choice`函数的基本语法如下：

numpy.random.choice(a, size=None, replace=True, p=None)

参数说明：

- `a`：一个一维数组或一个整数。如果是一个整数，那么它表示从`range(a)`中抽取样本。
- `size`：指定抽取样本的数量。可以是一个整数或者表示数组形状的元组。
- `replace`：一个布尔值，表示是否允许重复抽取。默认为`True`，即允许重复抽取。如果设置为`False`，那么抽取的样本将不重复。
- `p`：一个一维数组，表示抽取每个元素的概率。数组长度应与`a`相同。如果未指定，将使用均匀分布进行抽样。

在您的代码中：

batch = np.random.choice(mem_len, self.batch_size, replace=True)

`mem_len`可能是一个整数，表示从`range(mem_len)`中抽取样本；`self.batch_size`表示要抽取的样本数量；`replace=True`表示允许在抽取过程中重复选取相同的元素。所以，这行代码的作用是从`range(mem_len)`中随机抽取`self.batch_size`个样本，允许重复抽取。

万梓柠

isdkz 发表于 2023-4-21 11:27
`np.random.choice`是NumPy库中的一个函数，用于从给定的一维数组或整数范围中随机抽取指定数量的样本。`np ...

太强了太强了，膜拜大佬