Belle II Software development
custom.py
1#!/usr/bin/env python3
2
3
10
11
12def feature_importance(state):
13 """
14 Assert
15 """
16 raise RuntimeError("You have to overwrite feature_importance if you use the custom framework")
17
18
19def 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
26def load(obj):
27 """
28 Assert
29 """
30 raise RuntimeError("You have to overwrite load if you use the custom framework")
31
32
33def apply(state, X):
34 """
35 Assert
36 """
37 raise RuntimeError("You have to overwrite apply if you use the custom framework")
38
39
40def 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
47def 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
54def end_fit(state):
55 """
56 Assert
57 """
58 raise RuntimeError("You have to overwrite end_fit if you use the custom framework")