Python prepare_envの例

コード例 #1

    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()

コード例 #2

ファイル: test_sampling.py プロジェクト: zeta1999/graph-learn

  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()

コード例 #3

ファイル: test_edge.py プロジェクト: zymale/graph-learn

    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

コード例 #4

ファイル: test_node_query.py プロジェクト: knut0815/graph-learn

    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()

コード例 #5

ファイル: test_node.py プロジェクト: zymale/graph-learn

    def setUp(self):
        utils.prepare_env()

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