|
57 | 57 | "plt.legend()\n", |
58 | 58 | "plt.show()" |
59 | 59 | ] |
60 | | - }, |
61 | | - { |
62 | | - "cell_type": "code", |
63 | | - "execution_count": null, |
64 | | - "id": "ac7061fd", |
65 | | - "metadata": {}, |
66 | | - "outputs": [], |
67 | | - "source": [ |
68 | | - "sample_model=SampleModel(name='sample_model')\n", |
69 | | - "sample_model.temperature=5\n", |
70 | | - "sample_model.use_detailed_balance=True\n", |
71 | | - "sample_model.normalise_detailed_balance=True\n", |
72 | | - "\n", |
73 | | - "# Creating components\n", |
74 | | - "gaussian=Gaussian(name='Gaussian',width=0.5,area=1)\n", |
75 | | - "dho = DampedHarmonicOscillator(name='DHO',center=1.0,width=0.3,area=2.0)\n", |
76 | | - "lorentzian = Lorentzian(name='Lorentzian',center=-1.0,width=0.2,area=1.0)\n", |
77 | | - "polynomial = Polynomial(name='Polynomial',coefficients=[0.1, 0, 0.5]) # y=0.1+0.5*x^2\n", |
78 | | - "\n", |
79 | | - "sample_model.add_component(gaussian)\n", |
80 | | - "sample_model.add_component(dho)\n", |
81 | | - "sample_model.add_component(lorentzian)\n", |
82 | | - "sample_model.add_component(polynomial)\n", |
83 | | - "\n", |
84 | | - "\n", |
85 | | - "x=np.linspace(-2, 2, 100)\n", |
86 | | - "\n", |
87 | | - "plt.figure()\n", |
88 | | - "y=sample_model.evaluate(x)\n", |
89 | | - "plt.plot(x, y, label='Sample Model')\n", |
90 | | - "\n", |
91 | | - "for component in list(sample_model):\n", |
92 | | - " y = sample_model.evaluate_component(x, component.name)\n", |
93 | | - " plt.plot(x, y, label=component.name)\n", |
94 | | - "\n", |
95 | | - "plt.legend()\n", |
96 | | - "plt.title('Sample model at 5 K with detailed balance')\n", |
97 | | - "plt.show()" |
98 | | - ] |
99 | 60 | } |
100 | 61 | ], |
101 | 62 | "metadata": { |
|
0 commit comments