14 """ Constructor of the state object """
18 """ Returns representation of this state"""
22 def feature_importance(state):
24 Return a list containing the feature importances
26 print(
"Called feature importance")
30 def get_model(number_of_features, number_of_spectators, number_of_events, training_fraction, parameters):
32 Return default test model
34 print(
"Called get_model")
55 return np.require(p, dtype=np.float32, requirements=[
'A',
'W',
'C',
'O'])
58 def begin_fit(state, Xtest, Stest, ytest, wtest):
60 Test begin_fit function
62 print(
"Called begin_fit")
63 print(state, Xtest.shape, Stest.shape, ytest.shape, wtest.shape)
67 def partial_fit(state, X, S, y, w, epoch):
69 Test partial_fit function
71 print(
"Called partial_fit")
72 print(state, X.shape, S.shape, y.shape, w.shape, epoch)
80 print(
"Called end_fit")
82 return "SerializedState"