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)
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)