hiro/bandfit
1
0
Fork 0
mirror of https://github.com/vale981/bandfit synced 2025-03-05 09:31:42 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

adjust bounds

Browse source
This commit is contained in:
Valentin Boettcher 2023-03-21 14:28:34 -04:00
parent ed5eb15320
commit 9785bb5339
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
bandfit/bandfit.py
View file

@ -130,7 +130,7 @@ def candidate(k, c, d, a, δb, k_scale, k_shift):
def fit_to_bands( def fit_to_bands(
bands, bands,
bounds=[(-10, -10, 0.1, -0.5, 0.9, -0.5), (10, 10, 10, 0.5, 1.1, 0.5)], bounds=[(-10, -10, 0.1, -0.5, 0.8, -0.5), (10, 10, 10, 0.5, 1.2, 0.5)],
ic_scan_steps=5, ic_scan_steps=5,
c_d_order=0, c_d_order=0,
debug_plots=False, debug_plots=False,
@ -229,3 +229,13 @@ def plot_data_with_bands_and_fit(data, bands, band_fit):
plt.errorbar(ks, bands[:, 0], yerr=bands[:, 2]) plt.errorbar(ks, bands[:, 0], yerr=bands[:, 2])
plt.errorbar(ks, bands[:, 1], yerr=bands[:, 3]) plt.errorbar(ks, bands[:, 1], yerr=bands[:, 3])
plt.plot(smooth_ks_unscaled, upper_band) plt.plot(smooth_ks_unscaled, upper_band)
def plot_error_funnel(p, σ):
ks = np.linspace(-np.pi, np.pi, 1000)
params = np.random.multivariate_normal(p, np.diag(σ), 1000)
for param in params:
plt.plot(ks, energy(ks, *param), color="gray", alpha=0.1)
plt.plot(ks, energy(ks, *p), linewidth=2)