search

rpaudel42 / Pikachu

PIKACHU: Temporal Walk Based Dynamic Graph Embedding for Network Anomaly Detection
12 stars 4 forks source link

function predict_prob going extremely slow #3

Open alexis2804 opened 1 year ago

alexis2804 commented 1 year ago

Hello, I ran your code with the dataset sample given, it seems like the program is blocked at the function : 

def predict_prob(w, X):
    return softmax(np.dot(X, w.T))

The dot function is extremely slow and I can't figure out why. Which version of numpy did you use ?