Belle II Software  release-08-01-10
custom.py
1 #!/usr/bin/env python3
2 
3 
10 
11 
12 def feature_importance(state):
13  """
14  Assert
15  """
16  raise RuntimeError("You have to overwrite feature_importance if you use the custom framework")
17 
18 
19 def get_model(number_of_features, number_of_spectators, number_of_events, training_fraction, parameters):
20  """
21  Assert
22  """
23  raise RuntimeError("You have to overwrite get_model if you use the custom framework")
24 
25 
26 def load(obj):
27  """
28  Assert
29  """
30  raise RuntimeError("You have to overwrite load if you use the custom framework")
31 
32 
33 def apply(state, X):
34  """
35  Assert
36  """
37  raise RuntimeError("You have to overwrite apply if you use the custom framework")
38 
39 
40 def begin_fit(state, Xtest, Stest, ytest, wtest, nBatches):
41  """
42  Assert
43  """
44  raise RuntimeError("You have to overwrite begin_fit if you use the custom framework")
45 
46 
47 def partial_fit(state, X, S, y, w, epoch, batch):
48  """
49  Assert
50  """
51  raise RuntimeError("You have to overwrite partial_fit if you use the custom framework")
52 
53 
54 def end_fit(state):
55  """
56  Assert
57  """
58  raise RuntimeError("You have to overwrite end_fit if you use the custom framework")