Python model.Model类代码示例

def main():
    # NOTE: Cutoff can either be a single integer or it
    # can be a dictionary where the keys are two-tuples
    # of atomic numbers (e.g. (40,13)=3.5 for Zr,Al).
    modelfile = sys.argv[1]
    submodelfile = sys.argv[2]
    rotatedsubmodelfile = sys.argv[3]
    m = Model(modelfile)
    try:
        cut = 3.5  # float(sys.argv[2])
        cutoff = {}
        for z1 in m.atomtypes:
            for z2 in m.atomtypes:
                cutoff[(z1, z2)] = cut
                cutoff[(z2, z1)] = cut
    except:
        print("You didn't input a cutoff so you much define it in the code.")
    voronoi_3d(m, cutoff)

    subm = Model(submodelfile)
    rotsubm = Model(rotatedsubmodelfile)
    for atom in subm.atoms:
        if atom in m.atoms:
            rotsubm.atoms[atom.id].vp = m.atoms[m.atoms.index(atom)].vp
        else:
            print("Couldn't find atom {0} in full model!".format(atom))
    icofrac(rotsubm)
    # rotsubm.write_real_xyz()
    rotsubm.write_real_xyz(rotatedsubmodelfile[:-4] + ".icofrac.real.xyz")

    def post(self):
        id = self.get_argument("id",None)
        keyword = self.get_argument("keyword","")
        handler = Model('LvsManagerConfig')
        vipinstanceinfo = handler.getLvsManagerConfigVipInstanceList(id)

        def build(vipinstanceinfo):
            for row in vipinstanceinfo:
                for k,v in row.items():
                    if k=='vip_group':
                        rs=[]
                        for r in v:
                            rs.append(r['vip']+':'+r['port'])
                        row[k]="<br>".join(rs)
                    else:
                        if isinstance(v,unicode):
                            row[k]=str(v.encode('utf-8'))
                        else:
                            row[k]=str(v)
            return  json.dumps(vipinstanceinfo)

        result=[]
        for row in vipinstanceinfo:
            if str(row['descript']).find(keyword)!=-1:
                result.append(row)
        ret=""
        if len(result)>0:
            ret=build(result)
        else:
            ret=build(vipinstanceinfo)
        self.write(ret)

 def __init__(self, tracks=None, devices=None, transport=None,
                    view_scale=None, units=None, patch_bay=None):
   Model.__init__(self)
   # the file path to save to
   self.path = None
   # transport
   if (transport is None):
     transport = Transport()
   self.transport = transport
   # time scale
   if (view_scale is None):
     view_scale = ViewScale()
   self.view_scale = view_scale
   # devices
   if (devices is None):
     devices = DeviceAdapterList()
   self.devices = devices
   # a list of units on the workspace
   if (units is None):
     units = UnitList()
   self.units = units
   self.units.add_observer(self.on_change)
   self.units.add_observer(self.update_transport_duration)
   self.update_transport_duration()
   # a list of connections between units
   if (patch_bay is None):
     patch_bay = PatchBay()
   self.patch_bay = patch_bay
   self.patch_bay.add_observer(self.on_change)

def login(name,password):
    '''

    :param name:
    :param password:
    :return:  {"status":1, "username":user,"is_manager":False,"is_super_manager":False}
    '''
    try:
        ret=requests.post(API_URL+'login', {'user_name':name,'password':password}).json()
        if ret['status']=='1':
            user= ret['username']
            if user:
                handler = Model('Account')
                _find_user_result = handler.getAccountOne(user)
            if  _find_user_result:
                time_now = timestamptodate(time.time())
                handler.UpdateAccountPrivilege(user,ret['is_manager'],ret['is_super_manager'])
                handler.updateAccountLogintime(user,time_now)
            else:
                time_now = timestamptodate(time.time())
                user_data = {"username":user,"is_manager":ret['is_manager'],"is_super_manager":ret['is_super_manager'],"login_time":time_now,"register_time":time_now}
                handler.InsertAccount(user_data)
            return 1
        else:
            return 0
    except Exception as er:

        return 0

 def get(self):
     '''
     show publish.html page
     '''
     handler = Model('LvsPublish')
     result = handler.getLvsPublish()
     self.render2('publish.html',publish=result)

    def test_small_model(self):
        # Create one input
        X = helper.make_tensor_value_info('IN', TensorProto.FLOAT, [2, 3])
        # Create one output
        Y = helper.make_tensor_value_info('OUT', TensorProto.FLOAT, [2, 3])
        # Create a node
        node_def = helper.make_node('Abs', ['IN'], ['OUT'])

        # Create the model
        graph_def = helper.make_graph([node_def], "test-model", [X], [Y])
        onnx_model = helper.make_model(graph_def,
                                       producer_name='onnx-example')

        model = Model()
        model.BuildFromOnnxModel(onnx_model)
        schedule = model.OptimizeSchedule()
        schedule = schedule.replace('\n', ' ')
        expected_schedule = r'// Target: .+// MachineParams: .+// Delete this line if not using Generator Pipeline pipeline = get_pipeline\(\);.+Func OUT = pipeline.get_func\(1\);.+{.+}.+'
        self.assertRegex(schedule, expected_schedule)

        input = np.random.rand(2, 3) - 0.5
        outputs = model.run([input])
        self.assertEqual(1, len(outputs))
        output = outputs[0]
        expected = np.abs(input)
        np.testing.assert_allclose(expected, output)

    def initialize(self,inputParameterValues):

        # Inputs
        self.windVelNED                     = mat('0.0;0.0;0.0')

        # Received via message
        self.altitude           = 0.0
        self.dcmBodyFromNED     = mat([[0.,0.,0.],[0.,0.,0.],[0.,0.,0.]])
        self.velECEFinBody      = mat('0.;0.;0.')

        # Internal
        self.vwindmag                       = 0.0
        self.speedofsound                   = 0.0

        # Outputs
        self.mach                           = 0.0
        self.alpha                          = 0.0
        self.beta                           = 0.0
        self.density                        = 0.0
        self.pressure                       = 0.0
        self.temperature                    = 0.0
        self.uvw                            = mat('0.0;0.0;0.0')
        self.qbar                           = 0.0
        self.qalpha                         = 0.0

        # Call the base class initialize function, which sets all the input params
        Model.initialize(self,inputParameterValues)

def calc_entropy(K=20):
    global grammar
    global NORMALIZE
    global VERBOSE
    global start_cat
    cat_priors = Model('Prior')
    for cat in start_cat:
      cat_priors[cat] = 1
    normalize(cat_priors)
    for k in range(0,K):
        if VERBOSE:
          sys.stderr.write('Iterating ('+str(k)+'/'+str(K)+'): ')
          sys.stderr.write(str(cat_priors.keys())+'\n')
        tot = 0.0
        if NORMALIZE:
            normalize(cat_priors)
        cat_likes = Model('MarginL') #actually the new prior built from likelihoods
        for parent in cat_priors:
            for child in grammar[parent]:
                cat_likes[child] += grammar[parent][child]*cat_priors[parent]/3.0
        cat_priors = cat_likes.copy()

    # sum over the probability of each category to obtain the overall likelihood of the
    grammar_prob = 0.0
    for cat in cat_priors:
        grammar_prob += cat_priors[cat]
    
    # output the entropy of the grammar
    sys.stderr.write(str(-math.log(grammar_prob))+'\n')
    sys.stdout.write(str(-math.log(grammar_prob))+'\n')

def run(num_agents, num_items, prefs, dup_values):

    # Randomly generate some data for N agents and M items
    if prefs.dist_type == DistTypes.urand_int:
        m = Model.generate_urand_int(num_agents, num_items, dup_values)
    elif prefs.dist_type == DistTypes.urand_real:
        m = Model.generate_urand_real(num_agents, num_items, dup_values)
    elif prefs.dist_type == DistTypes.zipf_real:
        m = Model.generate_zipf_real(num_agents, num_items, 2., dup_values)
    elif prefs.dist_type == DistTypes.polya_urn_real:
        m = Model.generate_polya_urn_real(num_agents, num_items, 2, 1)
    elif prefs.dist_type == DistTypes.correlated_real:
        m = Model.generate_correlated_real(num_agents, num_items)
    else:
        raise Exception("Distribution type {0} is not recognized.".format(prefs.dist_type))

    # Do our bounding at the root to check for naive infeasibility
    #is_possibly_feasible, bounding_s = bounds.max_contested_feasible(m)
    #if not is_possibly_feasible:
    #    print "Bounded infeasible!"
    #    sys.exit(-1)

    # Compute an envy-free allocation (if it exists)
    stats = allocator.allocate(m, prefs)
    
    return stats

    def __init__(self, topology, bead_repr=None):
        """Structure-based Model (SBM)

        Parameters
        ----------
        topology : mdtraj.Topology object
            An mdtraj Topology object that describes the molecular topology.

        bead_repr : str [CA, CACB]
            A code specifying the desired coarse-grain mapping. The all-atom
        to coarse-grain mapping.

        Methods
        -------
        assign_* :
            Methods assign which atoms have bonded constraints
        (angle potentials, dihedral, etc.)

        add_* :
            Methods add potentials to the Hamiltonian.

        """



        Model.__init__(self, topology, bead_repr=bead_repr)
        self.Hamiltonian = potentials.StructureBasedHamiltonian()
        self.mapping.add_atoms()

def predict(args):
    with open(os.path.join(args.save_dir, 'config.pkl'), 'rb') as f:
        saved_args = pickle.load(f)
    with open(os.path.join(args.save_dir, 'chars_vocab.pkl'), 'rb') as f:
        chars, vocab = pickle.load(f)
    with open(os.path.join(args.save_dir, 'labels.pkl'), 'rb') as f:
        labels = pickle.load(f)

    model = Model(saved_args, deterministic=True)

    with open(args.data_path, 'r') as f:
        reader = csv.reader(f)
        texts = list(reader)

    texts = map(lambda i: i[0], texts)
    x = map(lambda i: transform(i.strip().decode('utf8'), saved_args.seq_length, vocab), texts)

    with tf.Session() as sess:
        saver =tf.train.Saver(tf.all_variables())
        ckpt = tf.train.get_checkpoint_state(args.save_dir)
        if ckpt and ckpt.model_checkpoint_path:
            saver.restore(sess, ckpt.model_checkpoint_path)

        start = time.time()
        results = model.predict_label(sess, labels, x)
        end = time.time()
        print 'prediction costs time: ', end - start

    with open(args.result_path, 'w') as f:
        writer = csv.writer(f)
        writer.writerows(zip(texts, results))

def main():
    modelfile = sys.argv[1]
    m = Model(modelfile)

    # Below is the identity rotation array.
    #rot_arr = [1,0,0,0,1,0,0,0,1]

    # Below is the correct rotation matrix for JWH t3 icofrac to get it into the orientation in his PRL paper.
    #rot_arr = [-0.031777, 0.998843, 0.036102, 0.986602, 0.025563, 0.161133, 0.160023, 0.040739, -0.986272]

    # Below is a (the?) rotation matrix for JWH t1 icofrac.
    #rot_arr = [ 0.954646, -0.233932, 0.184194, 0.280650, 0.913581, -0.294287, -0.099433, 0.332633, 0.937800 ]

    # Below is a (the?) rotation matrix of Pei's t1 that gives some planes. Oriented for a specific plane ~.
    #rot_arr = [ -0.977103, -0.123352, -0.173361, -0.130450, 0.990997, 0.030118, 0.168085, 0.052043, -0.984398 ]

    # Below is a (the?) rotation matrix of Pei's t2 that gives some planes. Oriented for a specific plane ~.
    rot_arr = [ 0.985478, -0.010230, -0.169493, 0.009247, 0.999936, -0.006586, 0.169549, 0.004923, 0.985509]

    # Below is a (the?) rotation matrix of Pei's t3 that gives some planes. Oriented for a specific plane ~.
    #rot_arr = [0.981624,-0.002765,-0.190808, -0.003436,0.999477,-0.032163, 0.190797,0.032228,0.981100]

    npra = np.asarray(rot_arr)
    rot(m,npra)

    # Write cif file to screen
    #m.write_cif()
    m.write_our_xyz()

    def update_ststc_customer_shop(self):
        query = """REPLACE INTO ststc_customer_shop(seller_nick, buyer_nick, costs, buy_times, last_buy_time)
SELECT  seller_nick,
        buyer_nick,
        SUM(costs) AS costs,
        SUM(buy_times) AS buy_times,
        MAX(last_buy_time) AS last_buy_time
FROM
    ((SELECT
        r.nick AS seller_nick,
        c.nick AS buyer_nick,
        SUM(r.auctionPrice) AS costs,
        COUNT(r.uid) AS buy_times,
        MAX(`date`) AS last_buy_time
    FROM rate_list AS r LEFT JOIN customer AS c ON r.uid = c.uid
    WHERE c.nick=%s)
    UNION
    (SELECT
        r.nick AS seller_nick,
        c.nick AS buyer_nick,
        SUM(r.price) AS costs,
        COUNT(r.uid) AS buy_times,
        MAX(`date`) AS last_buy_time
    FROM rate_list_past AS r LEFT JOIN customer AS c ON r.uid = c.uid
    WHERE c.nick=%s)) AS r"""
        Model.execute_db(query, self.nick, self.nick)

def sim_channels_2_model(sim_data, sim_key, group_key,):
    
    # instantiate an empty model
    model = Model(sim_key, objectives = [], meta = {})
    #debug_p('model id ' + str(model.get_model_id()))
    #debug_p('model num objectives before adding obj' + str(len(model.get_objectives())))
    
    # add meta data to model
    model.set_meta({'sim_id': sim_data['sim_id'], 'sim_meta':sim_data['meta'], 'sim_key':sim_key, 'group_key':group_key})

    
    for channel_code in objectives_channel_codes:
        
        # add model objective
        m_points = []
        for sample_point in channels_sample_points[channel_code]:
            m_points.append(sim_data[channel_code][sample_point])
        
        # assuming equal weights of objectives
        #debug_p('adding obj to simulation ' + str(sim_key))
        model.add_objective(channel_code, m_points)
        
    #debug_p('model num objectives after adding obj' + str(len(model.get_objectives())))
    
    return model

 def get_clusters_with_n_numneighs(self,cutoff,numneighs,cluster_types):
     m = Model(self.model.comment, self.model.lx, self.model.ly, self.model.lz, self.model.atoms[:])
     m.generate_neighbors(cutoff)
     vp_atoms = []
     #print(cluster_types)
     neighs = [[]]*m.natoms
     for i,atom in enumerate(m.atoms):
         if(atom.vp.type in cluster_types):
             vp_atoms.append(atom.copy())
     numfound = 0
     for i,atomi in enumerate(vp_atoms):
         for j,atomj in enumerate(vp_atoms[vp_atoms.index(atomi)+1:]):
             # Get all the neighbors they have in common
             #common_neighs = [n for n in atomi.neighs if n in atomj.neighs if n.vp.type not in cluster_types]
             common_neighs = [n for n in atomi.neighs if n in atomj.neighs]
             if(len(common_neighs) >= numneighs):
                 ind = m.atoms.index(atomi)
                 neighs[ind] = neighs[ind] + copy.copy([x for x in common_neighs if x not in neighs[ind]])
                 ind = m.atoms.index(atomj)
                 neighs[ind] = neighs[ind] + copy.copy([x for x in common_neighs if x not in neighs[ind]])
                 for n in common_neighs:
                     ind = m.atoms.index(n)
                     neighs[ind] = neighs[ind] + [x for x in [atomi,atomj] if x not in neighs[ind]]
                 numfound += 1
     for i,tf in enumerate(neighs):
         m.atoms[i].neighs = tf
     m.check_neighbors()
     print('Total number of {0} atoms: {1}'.format(cluster_types,len(vp_atoms)))
     print('Total number of {2}-sharing {0} atoms: {1}'.format(cluster_types,numfound,numneighs))
     # Now I should be able to go through the graph/model's neighbors.
     return self.search(m,cluster_types)