htmd.adaptive.adaptivebandit module#
- class htmd.adaptive.adaptivebandit.AdaptiveBandit#
Bases:
AdaptiveBase
- Parameters:
app (
SimQueue
object, default=None) – A SimQueue class object used to retrieve and submit simulationsproject (str, default='adaptive') – The name of the project
nmin (int, default=0) – Minimum number of running simulations
nmax (int, default=1) – Maximum number of running simulations
nepochs (int, default=1000) – Stop adaptive once we have reached this number of epochs
nframes (int, default=0) – Stop adaptive once we have simulated this number of aggregate simulation frames.
inputpath (str, default='input') – The directory used to store input folders
generatorspath (str, default='generators') – The directory containing the generators
dryrun (boolean, default=False) – A dry run means that the adaptive will retrieve and generate a new epoch but not submit the simulations
updateperiod (float, default=0) – When set to a value other than 0, the adaptive will run synchronously every updateperiod seconds
coorname (str, default='input.coor') – Name of the file containing the starting coordinates for the new simulations
lock (bool, default=False) – Lock the folder while adaptive is ongoing
datapath (str, default='data') – The directory in which the completed simulations are stored
filter (bool, default=True) – Enable or disable filtering of trajectories.
filtersel (str, default='not water') – Filtering atom selection
filteredpath (str, default='filtered') – The directory in which the filtered simulations will be stored
projection (
Projection
object, default=None) – A Projection class object or a list of objects which will be used to project the simulation data before constructing a Markov modelgoalfunction (function, default=None) – This function will be used to convert the goal-projected simulation data to a ranking whichcan be used for the directed component of FAST.
reward_method (str, default='max') – The reward method
statetype (('cluster', 'micro', 'macro'), str, default='micro') – State type (cluster, micro, macro) to use for reward calculations.
skip (int, default=1) – Allows skipping of simulation frames to reduce data. i.e. skip=3 will only keep every third frame
lag (int, default=1) – The lagtime used to create the Markov model. Units are in frames.
exploration (float, default=0.5) – Exploration is the coefficient used in UCB algorithm to weight the exploration value
temperature (int, default=300) – Temperature used to compute the free energy
ticalag (int, default=20) – Lagtime to use for TICA in frames. When using skip remember to change this accordinly.
ticadim (int, default=3) – Number of TICA dimensions to use. When set to 0 it disables TICA
clustmethod (
ClusterMixin
class, default=<class ‘sklearn.cluster._kmeans.MiniBatchKMeans’>) – Clustering algorithm used to cluster the contacts or distancesmacronum (int, default=8) – The number of macrostates to produce
save (bool, default=False) – Save the model generated
save_qval (bool, default=False) – Save the Q(a) and N values for every epoch
actionspace (('metric', 'goal', 'tica', 'ticapcca'), str, default='tica') – The projected space which will be clustered and used to define our action space
recluster (bool, default=False) – If to recluster the projected space used in MSM estimation with a different clustering.
reclusterMethod (, default=<class 'htmd.clustering.regular.RegCluster'>) – Clustering method for reclustering.
goal_init (float, default=0.3) – The proportional ratio of goal initialization compared to max frames set by nframes
- conformationStationaryDistribution(model)#
- count_pucb(q_value, exploration, predictor, step, n_value)#
- count_ucb(q_value, exploration, step, n_value)#
- getRewards(trajidx, data_q, confstatdist, numstates, rewardmethod)#