test.py

#!/usr/bin/env python3
 
 
 
import numpy as np
 
 
class State:
    """
    Test state
    """
 
    def __init__(self):
        """ Constructor of the state object """
 
    def __repr__(self):
        """ Returns representation of this state"""
        return 'State'
 
 
def feature_importance(state):
    """
    Return a list containing the feature importances
    """
    print("Called feature importance")
    return []
 
 
def get_model(number_of_features, number_of_spectators, number_of_events, training_fraction, parameters):
    """
    Return default test model
    """
    print("Called get_model")
    print(parameters)
    return State()
 
 
def load(obj):
    """
    Test load function
    """
    print("Called load")
    print(obj)
    return State()
 
 
def apply(state, X):
    """
    Test apply function
    """
    print("Called apply")
    print(state, X.shape)
    p = np.zeros(len(X))
    return np.require(p, dtype=np.float32, requirements=['A', 'W', 'C', 'O'])
 
 
def begin_fit(state, Xtest, Stest, ytest, wtest, nBatches):
    """
    Test begin_fit function
    """
    print("Called begin_fit")
    print(state, Xtest.shape, Stest.shape, ytest.shape, wtest.shape, nBatches)
    return state
 
 
def partial_fit(state, X, S, y, w, epoch, batch):
    """
    Test partial_fit function
    """
    print("Called partial_fit")
    print(state, X.shape, S.shape, y.shape, w.shape, epoch, batch)
    return True
 
 
def end_fit(state):
    """
    Test end_fit function
    """
    print("Called end_fit")
    print(state)
    return "SerializedState"