Skip to content

Commit

Permalink
Adapt intro notebook evaluator usage
Browse files Browse the repository at this point in the history
  • Loading branch information
opcode81 committed Sep 11, 2023
1 parent 5ecddef commit d7b4afc
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions notebooks/intro.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -250,7 +250,7 @@
"metadata": {},
"outputs": [],
"source": [
"evaluator_params = sensai.evaluation.VectorClassificationModelEvaluatorParams(data_splitter=data_splitter, compute_probabilities=True)\n",
"evaluator_params = sensai.evaluation.ClassificationEvaluatorParams(data_splitter=data_splitter, compute_probabilities=True)\n",
"evaluator = sensai.evaluation.VectorClassificationModelEvaluator(iris_io_data, params=evaluator_params)"
]
},
Expand Down Expand Up @@ -356,12 +356,12 @@
"metadata": {},
"outputs": [],
"source": [
"evaluatorParams = sensai.evaluation.VectorClassificationModelEvaluatorParams(\n",
"evaluatorParams = sensai.evaluation.ClassificationEvaluatorParams(\n",
" data_splitter=data_splitter, compute_probabilities=True,\n",
" additional_metrics=[sensai.eval_stats_classification.ClassificationMetricTopNAccuracy(2)])\n",
"cross_validator_params = sensai.evaluation.crossval.VectorModelCrossValidatorParams(folds=10,\n",
" evaluator_params=evaluator_params)\n",
"eval_util = sensai.evaluation.ClassificationEvaluationUtil(iris_io_data,\n",
"eval_util = sensai.evaluation.ClassificationModelEvaluation(iris_io_data,\n",
" evaluator_params=evaluatorParams, cross_validator_params=cross_validator_params)"
]
},
Expand Down

0 comments on commit d7b4afc

Please sign in to comment.