Source code for nlpatl.sampling.uncertainty.mismatch

from typing import Tuple, List, Union
from scipy.stats import entropy
import numpy as np

from nlpatl.sampling import Sampling


class MismatchSampling(Sampling):
	"""
		Sampling data points according to the mismatch. Pick the N data points
			randomly.

		:param name: Name of this sampling
		:type name: str
    """

	def __init__(self, name: str = 'mismatch_sampling'):
		super().__init__(name=name)

	def sample(self, data1: Union[List[str], List[int], List[float], np.ndarray],
		data2: Union[List[str], List[int], List[float], np.ndarray], 
		num_sample: int) -> Tuple[np.ndarray, np.ndarray]:

		assert len(data1) == len(data2), 'Two list of data have different size.'

		# Find mismatch
		mismatch_indices = []
		for i, (d1, d2) in enumerate(zip(data1, data2)):
			if d1 != d2:
				mismatch_indices.append(i)

		num_node = min(num_sample, len(mismatch_indices))

		mismatch_indices = np.random.choice(
			mismatch_indices, num_node, replace=False)

		return mismatch_indices, None