Пример #1
0
# test for Rabi flop fits

from model_test import ModelTest
from fit_rabi import Rabi

import numpy as np

test = ModelTest(Rabi, 'Rabi')
true_params = [2 * np.pi / (10), 10, 0.05, 0., 0, 0.6]
test.generate_data(0, 30, 300, 0.02, true_params)
test.fit()
test.print_results()
test.plot(fit=True)
# test for Gaussian fits

from model_test import ModelTest
from fit_gaussian import Gaussian

test = ModelTest(Gaussian, 'Gaussian')
true_params = [130., 4., 5., 0.1]
test.generate_data(100, 200, 200, 0.02, true_params)
test.fit()
test.print_results()
test.plot()