search

AnacletoLAB / grape

🍇 GRAPE is a Rust/Python Graph Representation Learning library for Predictions and Evaluations
MIT License
502 stars 38 forks source link

Edge prediction models not working in edge prediction evaluation pipeline #34

Open DylanSteinecke opened 1 year ago

DylanSteinecke commented 1 year ago

When using the edge prediction evaluation pipeline, I tried running a few different edge prediction models (e.g., KipfGCNEdgePrediction, not for the MLP, Decision Tree, or Random Forest), I got this error:

Traceback (most recent call last): File [script I ran] \ KipfGCNEdgePrediction(), File "/home/dylansteinecke/anaconda3/lib/python3.9/site-packages/embiggen/utils/abstract_models/model_stub.py", line 101, in init super().init(**parent_class.smoke_test_parameters()) File "/home/dylansteinecke/anaconda3/lib/python3.9/site-packages/embiggen/utils/abstract_models/abstract_model.py", line 124, in smoke_test_parameters raise NotImplementedError(( NotImplementedError: The smoke_test_parameters method must be implemented in the child classes of abstract model.

Is there a way I can address it or are these models not currently permitted in the edge prediction evaluation pipeline? Thank you.

LucaCappelletti94 commented 1 year ago

I think that is an error caused by a combination of errors, do you have TensorFlow and sklearn installed? I am on discord if you want to do a quick call.