search

bkj / graphsage-age-prediction

GraphSAGE Example: Age Prediction in POKEC social network
1 stars 1 forks source link

problem.h5 #1

Open ankur6ue opened 5 years ago

ankur6ue commented 5 years ago

Hi,

I want to step through the training code to see how it works. Downloaded the data and ran prep, but it gives me a bunch of exceptions. It does create the problem.h5 (558MB), but when I load the file in train, I get a 'folds' does not exist. I'm using Pytorch 1.01 and newer versions of all the libs, so likely some parsing error somewhere. May just be easier if you can post your problem.h5 file?

bkj commented 5 years ago

I'd guess it's a version problem -- I wouldn't expect this to run w/ the newest versions of all the libs. I'd make a conda env w/ the versions this was developed against, and see if you run into the same problems.

ankur6ue commented 5 years ago

Thanks. The prep script can be fixed (brought up to date) by these two changes:

  1. line 108: "folds" : np.array(folds, dtype=h5py.special_dtype(vlen=str))
  2. line 27: with pd.option_context('mode.use_inf_as_null', True): ages = ages.dropna()

The training script still crashes, if I find a fix, I'll report back here.