Searched refs:num_user (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_bgcf.py | 94 num_user, argument
98 …self.user_embed = Parameter(initializer("XavierUniform", [num_user, input_dim], dtype=mstype.float…
108 (self.input_dim, self.num_user, self.num_item) = dataset_argv
199 num_user, num_item = 7068, 3570
200 network = BGCF([64, num_user, num_item], 64, "tanh",
201 [0.0, 0.0, 0.0], num_user, num_item, 64)
204 users = Tensor(np.zeros([num_user,]).astype(np.int32))
207 u_test_neighs = Tensor(np.zeros([num_user, 40]).astype(np.int32))
208 u_test_gnew_neighs = Tensor(np.zeros([num_user, 20]).astype(np.int32))
|