Skip to content

Instantly share code, notes, and snippets.

View dainis-boumber's full-sized avatar
🏠
Working from home

Dainis Boumber dainis-boumber

🏠
Working from home
View GitHub Profile
import os
import re
import string
import numpy as np
from sklearn.feature_extraction.text import CountVectorizer
import torch
from torch import nn
from __future__ import division
import numpy as np
def mut_info(X, y):
n = X.shape[0]
n_11 = np.asarray(X[y].sum(axis=0)).squeeze()
n_01 = np.asarray(X[~y].sum(axis=0)).squeeze()
n_10 = np.asarray(y.sum() - n_11).squeeze()
n_00 = np.asarray(n - n_11 - n_01 - n_10).squeeze()