A RetroSearch Logo

Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Search Query:

Showing content from https://github.com/JuliaML/MLDatasets.jl/issues/223 below:

OGBDataset ogbg-molhiv has wrong shape of edge features · Issue #223 · JuliaML/MLDatasets.jl · GitHub

Wrong dimension of edge features, there are 6 edge features, but 3 should be correct

using MLDatasets, DataFrames

data = OGBDataset("ogbg-molhiv")
size(data[1].graphs.edge_data.features)

returns

However,

returns

I assume the correct size of the edge features would be (3,40) instead of (6,20)

I am using DataFrames v1.6.1 and MLDatasets v0.7.14

Using python and obg, the shape is (40,3):

from ogb.graphproppred import PygGraphPropPredDataset
dataset = PygGraphPropPredDataset(name = "ogbg-molhiv", root = '.')
dataset[0].edge_attr.shape
torch.Size([40, 3])

and

dataset[0].edge_index.shape
torch.Size([2, 40])

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.4