Ejemplo n.º 1
0
    def setUpClass(cls):
        gl.set_default_neighbor_id(-1)
        gl.set_default_int_attribute(1000)
        gl.set_default_float_attribute(999.9)
        gl.set_default_string_attribute('hehe')

        utils.prepare_env()
Ejemplo n.º 2
0
  def setUpClass(cls):
    gl.set_default_neighbor_id(-1)
    gl.set_default_int_attribute(1000)
    gl.set_default_float_attribute(999.9)
    gl.set_default_string_attribute('hehe')
    gl.set_padding_mode(gl.REPLICATE) # default

    utils.prepare_env()
Ejemplo n.º 3
0
    def setUp(self):
        utils.prepare_env()

        self.src_type_ = 'user'
        self.dst_type_ = 'item'
        self.edge_type_ = "first"
        self.edge_tuple_ = (self.src_type_, self.dst_type_, self.edge_type_)
        self.edge_type_list_ = ["first", "second"]
        self.src_range_ = (0, 100)
        self.dst_range_ = (100, 200)
        self.src_ids_ = np.array([2, 5, 8])
        self.dst_ids_ = np.array([102, 105, 108])
        self.batch_size_ = 3
Ejemplo n.º 4
0
    def setUpClass(cls):
        gl.set_default_int_attribute(10000)
        gl.set_default_float_attribute(0.01)
        gl.set_default_string_attribute('default')

        utils.prepare_env()
Ejemplo n.º 5
0
    def setUp(self):
        utils.prepare_env()

        self.node_type_ = 'user'
        self.value_range_ = [(0, 100)]
        self.ids_ = np.array([2, 5, 8])