dmlc--dgl
Hongzhi (Steve), Chen
f19f05ce66
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
11 行
272 B
Python
11 行
272 B
Python
import numpy as np
|
|
from sklearn.metrics import f1_score
|
|
|
|
|
|
# function to compute f1 score
|
|
def evaluate_f1_score(pred, label):
|
|
pred = np.round(pred, 0).astype(np.int16)
|
|
pred = pred.flatten()
|
|
label = label.flatten()
|
|
return f1_score(y_pred=pred, y_true=label)
|