HypTrails

This tutorial is still in development!

import numpy as np
from scipy.sparse import csr_matrix
from sklearn.preprocessing import normalize

from pytrails.hyptrails import *

transition_counts = csr_matrix(
    [[1, 2, 3],
     [4, 5, 6],
     [7, 8, 9]],
    dtype=np.float64)
transition_probabilities = normalize(transition_counts, "l1", axis=1)
pseudo_counts = transition_probabilities * 5
ml = MarkovChain.marginal_likelihood(transition_counts, pseudo_counts)
print(ml)

-51.7608370813