Link Prediction Framework in Python