Exemplo n.º 1
0
def saveModelData(filename,y_name,fixed_names,y,X,tree):
    headers =  [y_name] + fixed_names + ['species']
    (n,p) = np.shape(X)
    fixed = [np.asarray(X[:,i]).reshape(-1) for i in range(1,p)]
    data_columns = [y] + fixed + [tree.tipNames]
    #data_columns = [y] + [ X[,i] for i in range(1,w)]
    gt.createDataFile(filename,headers,data_columns)
Exemplo n.º 2
0
def saveModelData2(filename,y_name,fixed_names,random_names,y,X,Z,trees):
    headers =  [y_name] + fixed_names + random_names
    (n,p) = np.shape(X)
    # = [np.asarray(d).reshape(np.size(d)) for d in column_arrays]
    fixed = [np.asarray(X[:,i]).reshape(-1) for i in range(1,p)]
    # Match Z against each tree in trees
    random_columns = convertZ(Z,n,[tree.tipNames for tree in trees])
    # concatenate all data columns
    data_columns = [np.asarray(y).reshape(-1)] + fixed + random_columns
    #data_columns = [y] + [ X[,i] for i in range(1,w)]
    gt.createDataFile(filename,headers,data_columns)