traintest
evaluate_hold_out(model, fun_control)
¶
Evaluate a model using hold-out validation. A validation set is created from the training set. The test set is not used in this evaluation.
Note:
In contrast to evaluate_model()
, this function creates a validation set as
a subset of the training set.
It can be selected by setting fun_control["eval"] = "evaluate_hold_out"
.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
model |
sklearn model
|
sklearn model. |
required |
fun_control |
dict
|
dictionary containing control parameters for the function. |
required |
Returns:
Type | Description |
---|---|
ndarray
|
array containing evaluation results. |
Raises:
Type | Description |
---|---|
Exception
|
if call to train_test_split() or fit() or predict() fails. |
Source code in spotpython/sklearn/traintest.py
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
|
evaluate_model(model, fun_control)
¶
Evaluate a model using the test set.
First, the model is trained on the training set. If a scaler
is provided, the data is transformed using the scaler and fit_transform(X_train)
.
Then, the model is evaluated using the test set from fun_control
,
the scaler with transform(X_test)
,
the model.predict() method and the
metric_params
specified in fun_control
.
Note:
In contrast to evaluate_hold_out()
, this function uses the test set.
It can be selected by setting fun_control["eval"] = "eval_test"
.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
model |
sklearn model
|
sklearn model. |
required |
fun_control |
dict
|
dictionary containing control parameters for the function. |
required |
Returns:
Type | Description |
---|---|
ndarray
|
array containing evaluation results. |
Source code in spotpython/sklearn/traintest.py
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
|
evaluate_model_oob(model, fun_control)
¶
Out-of-bag evaluation (Only for RandomForestClassifier). If fun_control[“eval”] == “eval_oob_score”.
Source code in spotpython/sklearn/traintest.py
132 133 134 135 136 137 138 139 140 141 142 143 144 145 |
|