TfData.py

#!/usr/bin/env python
 

 
 
import numpy as np
 
 
class TfDataBasf2():
    """
    handles data, necessary for the training
    """
 
    def __init__(self, train_x, train_y, valid_x, valid_y, batch_size, seed=None, epoch_random_shuffle=True):
        """
        declaration of class variables
        """
        
        self.train_x = train_x.astype(np.float32)
        
        self.train_y = train_y.astype(np.float32)
        
        self.valid_x = valid_x.astype(np.float32)
        
        self.valid_y = valid_y.astype(np.float32)
 
        
        self.batch_size = batch_size
        
        self.seed = seed
        
        self.epoch_random_shuffle = epoch_random_shuffle
 
        
        self.train_events = self.train_x.shape[0]
        
        self.valid_events = self.valid_x.shape[0]
 
        
        self.feature_number = self.train_x.shape[1]
 
        
        self.batches = self.train_x.shape[0] // self.batch_size
 
        
        self.train_idx = np.zeros(self.train_x.shape[0])
 
        
        self.batch_train_x = np.zeros((self.feature_number, self.batch_size))
 
        
        self.batch_train_y = np.zeros(self.batch_size)
 
        
        self.random_state = np.random.RandomState(seed)
 
        
        self.sanitize_labels()
 
    def sanitize_labels(self):
        """
        checks for a binary classification problem
        transforms the two class labels to {0,1}
        """
        # not binary
        assert len(np.unique(self.train_y)) == 2
        # different classes
        assert np.array_equal(np.unique(self.train_y), np.unique(self.valid_y))
 
        # reject corner cases when classes would have special values
        if self.train_y.min() > 0:
            self.train_y[self.train_y == self.train_y.min()] = 0
            self.valid_y[self.valid_y == self.valid_y.min()] = 0
 
        if self.train_y.max() != 1:
            self.train_y[self.train_y == self.train_y.max()] = 1
            self.valid_y[self.valid_y == self.valid_y.max()] = 1
 
        # transform labels
        if self.train_y.min() != 0:
            self.train_y[self.train_y == self.train_y.min()] = 0
            self.valid_y[self.valid_y == self.valid_y.min()] = 0
 
    def batch_iterator(self):
        """
        iterator to provide training batches
        """
        self.train_idx = np.arange(len(self.train_idx))
 
        if self.epoch_random_shuffle:
            self.random_state.shuffle(self.train_idx)
 
        for i in range(self.batches):
            self.batch_train_x = self.train_x[self.train_idx[i * self.batch_size: (i + 1) * self.batch_size]]
            self.batch_train_y = self.train_y[self.train_idx[i * self.batch_size: (i + 1) * self.batch_size]]
 
            yield self.batch_train_x, self.batch_train_y
 
 
class TfDataBasf2Stub():
    """
    stub class just for initializing in basf2 begin_run
    """
 
    def __init__(self, batch_size, feature_number, event_number, train_fraction):
        """
        declare for initialization required batch parameters
        """
        
        self.batch_size = batch_size
 
        
        self.feature_number = feature_number
 
        
        self.batches = (event_number * train_fraction) // self.batch_size
 
        
        self.train_events = int(train_fraction * event_number)
 
        
        self.valid_events = int((1 - train_fraction) * event_number)