--- title: Calibration keywords: fastai sidebar: home_sidebar summary: "Functionality to calibrate a trained, binary classification model using temperature scaling." description: "Functionality to calibrate a trained, binary classification model using temperature scaling." nb_path: "nbs/052c_calibration.ipynb" ---
from tsai.basics import *
from tsai.models.FCNPlus import FCNPlus
X, y, splits = get_UCR_data('FingerMovements', split_data=False)
tfms = [None, [TSClassification()]]
batch_tfms = TSRobustScale()
dls = get_ts_dls(X, y, splits=splits, tfms=tfms, batch_tfms=batch_tfms)
learn = ts_learner(dls, FCNPlus, metrics=accuracy)
learn.fit_one_cycle(2)
learn.calibrate_model()
calibrated_model = learn.calibrated_model