Page MenuHomec4science
Files F87350702

model.py
No OneTemporary
Actions

File Metadata

Created
Sat, Oct 12, 03:52

model.py
View Options

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author: Theo Lemaire
# @Date: 2017-08-03 11:53:04
# @Email: theo.lemaire@epfl.ch
# @Last Modified by: Theo Lemaire
# @Last Modified time: 2019-06-02 13:23:59
import pickle
import abc
import inspect
import numpy as np
from .batches import createQueue
from ..utils import logger
class Model(metaclass=abc.ABCMeta):
''' Generic model interface. '''
@property
@abc.abstractmethod
def tscale(self):
''' Relevant temporal scale of the model. '''
raise NotImplementedError
@property
@abc.abstractmethod
def __repr__(self):
raise NotImplementedError
def params(self):
''' Gather all model parameters in a dictionary '''
def toAvoid(p):
return (p.startswith('__') and p.endswith('__')) or p.startswith('_abc_')
class_attrs = inspect.getmembers(self.__class__, lambda a: not(inspect.isroutine(a)))
inst_attrs = inspect.getmembers(self, lambda a: not(inspect.isroutine(a)))
class_attrs = [a for a in class_attrs if not toAvoid(a[0])]
inst_attrs = [a for a in inst_attrs if not toAvoid(a[0]) and a not in class_attrs]
params_dict = {a[0]: a[1] for a in class_attrs + inst_attrs}
return params_dict
@property
@abc.abstractmethod
def filecode(self, *args):
raise NotImplementedError
def getDesc(self):
return inspect.getdoc(self).splitlines()[0]
@property
@abc.abstractmethod
def getPltScheme(self):
raise NotImplementedError
@property
@abc.abstractmethod
def getPltVars(self, *args, **kwargs):
raise NotImplementedError
@property
@abc.abstractmethod
def checkInputs(self, *args):
raise NotImplementedError
@property
@abc.abstractmethod
def meta(self, *args):
raise NotImplementedError
@property
@abc.abstractmethod
def simulate(self, *args, **kwargs):
raise NotImplementedError
def simQueue(self, *args):
''' Create a simulation queue from a combination of simulation parameters. '''
return createQueue(*args)
def runAndSave(self, outdir, *args):
''' Simulate system and save results in a PKL file. '''
# If no amplitude provided, perform titration to find it
if None in args:
iA = args.index(None)
new_args = [x for x in args if x is not None]
Athr = self.titrate(*new_args)
if np.isnan(Athr):
logger.error('Could not find threshold excitation amplitude')
return None
new_args.insert(iA, Athr)
args = new_args
# Simulate model, save inf file and return file path
data, tcomp = self.simulate(*args)
meta = self.meta(*args)
meta['tcomp'] = tcomp
outpath = '{}/{}.pkl'.format(outdir, self.filecode(*args))
with open(outpath, 'wb') as fh:
pickle.dump({'meta': meta, 'data': data}, fh)
logger.debug('simulation data exported to "%s"', outpath)
return outpath

Event Timeline

c4science · Help