diff --git a/examples/pytorch/metapath2vec/test.py b/examples/pytorch/metapath2vec/test.py index 3134c386d283..ca6ab0ae268f 100644 --- a/examples/pytorch/metapath2vec/test.py +++ b/examples/pytorch/metapath2vec/test.py @@ -94,9 +94,9 @@ author_true = np.array(author_true) file.close() print("beging predicting") - clf_venue = LogisticRegression( - random_state=0, solver="lbfgs" - ).fit(venue_training, venue_label) + clf_venue = LogisticRegression(random_state=0, solver="lbfgs").fit( + venue_training, venue_label + ) y_pred_venue = clf_venue.predict(venue_testing) clf_author = LogisticRegression(random_state=0, solver="lbfgs").fit( author_training, author_label