Skip to content
Snippets Groups Projects
Commit f20fae74 authored by Bas Nijholt's avatar Bas Nijholt
Browse files

update phasediagram code

parent d3081de5
No related branches found
No related tags found
No related merge requests found
Pipeline #21898 passed
Stage: test
Hide whitespace changes
Inline Side-by-side
phase_diagram.ipynb
+ 9
− 12
View file @ f20fae74
%% Cell type:code id: tags:
```
%%writefile learners_file.py
import phase_diagram
import adaptive
import phase_diagram
lead_pars = dict(
a=10, r1=50, r2=70, coverage_angle=135, angle=45, with_shell=True, which_lead=""
)
params = dict(
alpha=20,
B_x=0,
B_y=0,
B_z=0,
Delta=110,
g=50,
orbital=True,
mu_sc=100,
c_tunnel=3 / 4,
V_r=-50,
intrinsic_sc=False,
mu_=lambda x0, sigma, mu_lead, mu_wire: mu_lead,
V_=lambda z, V_0, V_r, V_l, x0, sigma, r1: 0,
V_0=None,
V_l=None,
mu_lead=10,
mu_wire=None,
r1=None,
sigma=None,
x0=None,
**phase_diagram.constants.__dict__
)
def pf(xy, params=params, lead_pars=lead_pars):
import phase_diagram
params["B_x"], params["mu_lead"] = xy
lead = phase_diagram.make_lead(**lead_pars).finalized()
return phase_diagram.calculate_pfaffian(lead, params)
def smallest_gap(xy, params=params, lead_pars=lead_pars):
import phase_diagram
params["B_x"], params["mu_lead"] = xy
lead = phase_diagram.make_lead(**lead_pars).finalized()
pf = phase_diagram.calculate_pfaffian(lead, params)
gap = phase_diagram.gap_from_modes(lead, params)
return pf * gap
fname = 'phase_diagram_gap.pickle'
f = smallest_gap
# fname = 'phase_diagram.pickle'
# f = pf
loss = adaptive.learner.learnerND.curvature_loss_function()
learner = adaptive.LearnerND(smallest_gap, bounds=[(0, 2), (0, 35)], loss_per_simplex=loss)
# learner.load(fname)
learners = [learner]
fnames = [fname]
```
%% Cell type:code id: tags:
```
import adaptive
adaptive.notebook_extension()
runner = adaptive.Runner(learner, goal=lambda l: l.npoints > 60000)
runner.live_info()
```
%% Cell type:code id: tags:
```
%%output size=100
learner.plot(tri_alpha=0.4, n=None)
```
%% Cell type:code id: tags:
```
learner.save(fname)
```
%% Cell type:code id: tags:
```
import adaptive_scheduler
def goal(learner):
return learner.npoints > 100_000
return learner.npoints > 200
scheduler = adaptive_scheduler.scheduler.PBS(
cores=10*20,
cores_per_node=20,
executor_type='ipyparallel',
) # every learner get this many cores
scheduler = adaptive_scheduler.scheduler.DefaultScheduler(
cores=100,
executor_type="ipyparallel",
) # PBS or SLURM
run_manager = adaptive_scheduler.server_support.RunManager(
scheduler=scheduler,
learners_file="learners_file.py",
goal=goal,
log_interval=30,
save_interval=120,
save_interval=30,
job_name='phase-diagram'
)
run_manager.start()
```
%% Cell type:code id: tags:
```
```
......
phase_diagram.py
+ 0
− 4
View file @ f20fae74
import inspect
import operator
import sys
......@@ -13,7 +11,6 @@ import numpy as np
import scipy.constants
from kwant.continuum.discretizer import discretize
import adaptive
import pfaffian as pf
assert sys.version_info >= (3, 6), "Use Python ≥3.6"
......@@ -373,7 +370,6 @@ def apply_peierls_to_template(template, xyz_offset=(0, 0, 0)):
def get_offset(shape, start, lat):
a = np.max(lat.prim_vecs)
coords = [site.pos for site in lat.shape(shape, start)()]
xyz_offset = np.mean(coords, axis=0)
return xyz_offset
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment