annotate MicroPITA.py @ 0:0de566f21448 draft default tip

v2
author sagun98
date Thu, 03 Jun 2021 18:13:32 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
sagun98
parents:
diff changeset
1 #!/usr/bin/env python
sagun98
parents:
diff changeset
2 """
sagun98
parents:
diff changeset
3 Author: Timothy Tickle
sagun98
parents:
diff changeset
4 Description: Class to Run analysis for the microPITA paper
sagun98
parents:
diff changeset
5 """
sagun98
parents:
diff changeset
6
sagun98
parents:
diff changeset
7 #####################################################################################
sagun98
parents:
diff changeset
8 #Copyright (C) <2012>
sagun98
parents:
diff changeset
9 #
sagun98
parents:
diff changeset
10 #Permission is hereby granted, free of charge, to any person obtaining a copy of
sagun98
parents:
diff changeset
11 #this software and associated documentation files (the "Software"), to deal in the
sagun98
parents:
diff changeset
12 #Software without restriction, including without limitation the rights to use, copy,
sagun98
parents:
diff changeset
13 #modify, merge, publish, distribute, sublicense, and/or sell copies of the Software,
sagun98
parents:
diff changeset
14 #and to permit persons to whom the Software is furnished to do so, subject to
sagun98
parents:
diff changeset
15 #the following conditions:
sagun98
parents:
diff changeset
16 #
sagun98
parents:
diff changeset
17 #The above copyright notice and this permission notice shall be included in all copies
sagun98
parents:
diff changeset
18 #or substantial portions of the Software.
sagun98
parents:
diff changeset
19 #
sagun98
parents:
diff changeset
20 #THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
sagun98
parents:
diff changeset
21 #INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
sagun98
parents:
diff changeset
22 #PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
sagun98
parents:
diff changeset
23 #HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
sagun98
parents:
diff changeset
24 #OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
sagun98
parents:
diff changeset
25 #SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
sagun98
parents:
diff changeset
26 #####################################################################################
sagun98
parents:
diff changeset
27
sagun98
parents:
diff changeset
28 __author__ = "Timothy Tickle"
sagun98
parents:
diff changeset
29 __copyright__ = "Copyright 2012"
sagun98
parents:
diff changeset
30 __credits__ = ["Timothy Tickle"]
sagun98
parents:
diff changeset
31 __license__ = "MIT"
sagun98
parents:
diff changeset
32 __maintainer__ = "Timothy Tickle"
sagun98
parents:
diff changeset
33 __email__ = "ttickle@sph.harvard.edu"
sagun98
parents:
diff changeset
34 __status__ = "Development"
sagun98
parents:
diff changeset
35
sagun98
parents:
diff changeset
36 import sys
sagun98
parents:
diff changeset
37 import argparse
sagun98
parents:
diff changeset
38 from src.breadcrumbs.src.AbundanceTable import AbundanceTable
sagun98
parents:
diff changeset
39 from src.breadcrumbs.src.ConstantsBreadCrumbs import ConstantsBreadCrumbs
sagun98
parents:
diff changeset
40 from src.breadcrumbs.src.Metric import Metric
sagun98
parents:
diff changeset
41 from src.breadcrumbs.src.KMedoids import Kmedoids
sagun98
parents:
diff changeset
42 from src.breadcrumbs.src.MLPYDistanceAdaptor import MLPYDistanceAdaptor
sagun98
parents:
diff changeset
43 from src.breadcrumbs.src.SVM import SVM
sagun98
parents:
diff changeset
44 from src.breadcrumbs.src.UtilityMath import UtilityMath
sagun98
parents:
diff changeset
45
sagun98
parents:
diff changeset
46 from src.ConstantsMicropita import ConstantsMicropita
sagun98
parents:
diff changeset
47 import csv
sagun98
parents:
diff changeset
48 import logging
sagun98
parents:
diff changeset
49 import math
sagun98
parents:
diff changeset
50 import mlpy
sagun98
parents:
diff changeset
51 import numpy as np
sagun98
parents:
diff changeset
52 import operator
sagun98
parents:
diff changeset
53 import os
sagun98
parents:
diff changeset
54 import random
sagun98
parents:
diff changeset
55 import scipy.cluster.hierarchy as hcluster
sagun98
parents:
diff changeset
56 import scipy.spatial.distance
sagun98
parents:
diff changeset
57 from types import *
sagun98
parents:
diff changeset
58
sagun98
parents:
diff changeset
59 class MicroPITA:
sagun98
parents:
diff changeset
60 """
sagun98
parents:
diff changeset
61 Selects samples from a first tier of a multi-tiered study to be used in a second tier.
sagun98
parents:
diff changeset
62 Different methods can be used for selection.
sagun98
parents:
diff changeset
63 The expected input is an abundance table (and potentially a text file of targeted features,
sagun98
parents:
diff changeset
64 if using the targeted features option). Output is a list of samples exhibiting the
sagun98
parents:
diff changeset
65 characteristics of interest.
sagun98
parents:
diff changeset
66 """
sagun98
parents:
diff changeset
67
sagun98
parents:
diff changeset
68 #Constants
sagun98
parents:
diff changeset
69 #Diversity metrics Alpha
sagun98
parents:
diff changeset
70 c_strInverseSimpsonDiversity = Metric.c_strInvSimpsonDiversity
sagun98
parents:
diff changeset
71 c_strChao1Diversity = Metric.c_strChao1Diversity
sagun98
parents:
diff changeset
72
sagun98
parents:
diff changeset
73 #Diversity metrics Beta
sagun98
parents:
diff changeset
74 c_strBrayCurtisDissimilarity = Metric.c_strBrayCurtisDissimilarity
sagun98
parents:
diff changeset
75
sagun98
parents:
diff changeset
76 #Additive inverses of diversity metrics beta
sagun98
parents:
diff changeset
77 c_strInvBrayCurtisDissimilarity = Metric.c_strInvBrayCurtisDissimilarity
sagun98
parents:
diff changeset
78
sagun98
parents:
diff changeset
79 #Technique Names
sagun98
parents:
diff changeset
80 ConstantsMicropita.c_strDiversity2 = ConstantsMicropita.c_strDiversity+"_C"
sagun98
parents:
diff changeset
81
sagun98
parents:
diff changeset
82 #Targeted feature settings
sagun98
parents:
diff changeset
83 c_strTargetedRanked = ConstantsMicropita.c_strTargetedRanked
sagun98
parents:
diff changeset
84 c_strTargetedAbundance = ConstantsMicropita.c_strTargetedAbundance
sagun98
parents:
diff changeset
85
sagun98
parents:
diff changeset
86 #Technique groupings
sagun98
parents:
diff changeset
87 # c_lsDiversityMethods = [ConstantsMicropita.c_strDiversity,ConstantsMicropita.c_strDiversity2]
sagun98
parents:
diff changeset
88
sagun98
parents:
diff changeset
89 #Converts ecology metrics into standardized method selection names
sagun98
parents:
diff changeset
90 dictConvertAMetricDiversity = {c_strInverseSimpsonDiversity:ConstantsMicropita.c_strDiversity, c_strChao1Diversity:ConstantsMicropita.c_strDiversity2}
sagun98
parents:
diff changeset
91 # dictConvertMicroPITAToAMetric = {ConstantsMicropita.c_strDiversity:c_strInverseSimpsonDiversity, ConstantsMicropita.c_strDiversity2:c_strChao1Diversity}
sagun98
parents:
diff changeset
92 dictConvertBMetricToMethod = {c_strBrayCurtisDissimilarity:ConstantsMicropita.c_strRepresentative}
sagun98
parents:
diff changeset
93 dictConvertInvBMetricToMethod = {c_strBrayCurtisDissimilarity:ConstantsMicropita.c_strExtreme}
sagun98
parents:
diff changeset
94
sagun98
parents:
diff changeset
95 #Linkage used in the Hierarchical clustering
sagun98
parents:
diff changeset
96 c_strHierarchicalClusterMethod = 'average'
sagun98
parents:
diff changeset
97
sagun98
parents:
diff changeset
98 ####Group 1## Diversity
sagun98
parents:
diff changeset
99 #Testing: Happy path Testing (8)
sagun98
parents:
diff changeset
100 def funcGetTopRankedSamples(self, lldMatrix = None, lsSampleNames = None, iTopAmount = None):
sagun98
parents:
diff changeset
101 """
sagun98
parents:
diff changeset
102 Given a list of lists of measurements, for each list the indices of the highest values are returned. If lsSamplesNames is given
sagun98
parents:
diff changeset
103 it is treated as a list of string names that is in the order of the measurements in each list. Indices are returned or the sample
sagun98
parents:
diff changeset
104 names associated with the indices.
sagun98
parents:
diff changeset
105
sagun98
parents:
diff changeset
106 :param lldMatrix: List of lists [[value,value,value,value],[value,value,value,value]].
sagun98
parents:
diff changeset
107 :type: List of lists List of measurements. Each list is a different measurement. Each measurement in positionally related to a sample.
sagun98
parents:
diff changeset
108 :param lsSampleNames: List of sample names positionally related (the same) to each list (Optional).
sagun98
parents:
diff changeset
109 :type: List of strings List of strings.
sagun98
parents:
diff changeset
110 :param iTopAmount: The amount of top measured samples (assumes the higher measurements are better).
sagun98
parents:
diff changeset
111 :type: integer Integer amount of sample names/ indices to return.
sagun98
parents:
diff changeset
112 :return List: List of samples to be selected.
sagun98
parents:
diff changeset
113 """
sagun98
parents:
diff changeset
114 topRankListRet = []
sagun98
parents:
diff changeset
115 for rowMetrics in lldMatrix:
sagun98
parents:
diff changeset
116 #Create 2 d array to hold value and index and sort
sagun98
parents:
diff changeset
117 liIndexX = [rowMetrics,range(len(rowMetrics))]
sagun98
parents:
diff changeset
118 liIndexX[1].sort(key = liIndexX[0].__getitem__,reverse = True)
sagun98
parents:
diff changeset
119
sagun98
parents:
diff changeset
120 if lsSampleNames:
sagun98
parents:
diff changeset
121 topRankListRet.append([lsSampleNames[iIndex] for iIndex in liIndexX[1][:iTopAmount]])
sagun98
parents:
diff changeset
122 else:
sagun98
parents:
diff changeset
123 topRankListRet.append(liIndexX[1][:iTopAmount])
sagun98
parents:
diff changeset
124
sagun98
parents:
diff changeset
125 return topRankListRet
sagun98
parents:
diff changeset
126
sagun98
parents:
diff changeset
127 ####Group 2## Representative Dissimilarity
sagun98
parents:
diff changeset
128 #Testing: Happy path tested 1
sagun98
parents:
diff changeset
129 def funcGetCentralSamplesByKMedoids(self, npaMatrix=None, sMetric=None, lsSampleNames=None, iNumberSamplesReturned=0, istmBetaMatrix=None, istrmTree=None, istrmEnvr=None):
sagun98
parents:
diff changeset
130 """
sagun98
parents:
diff changeset
131 Gets centroid samples by k-medoids clustering of a given matrix.
sagun98
parents:
diff changeset
132
sagun98
parents:
diff changeset
133 :param npaMatrix: Numpy array where row=features and columns=samples
sagun98
parents:
diff changeset
134 :type: Numpy array Abundance Data.
sagun98
parents:
diff changeset
135 :param sMetric: String name of beta metric used as the distance metric.
sagun98
parents:
diff changeset
136 :type: String String name of beta metric.
sagun98
parents:
diff changeset
137 :param lsSampleNames: The names of the sample
sagun98
parents:
diff changeset
138 :type: List List of strings
sagun98
parents:
diff changeset
139 :param iNumberSamplesReturned: Number of samples to return, each will be a centroid of a sample.
sagun98
parents:
diff changeset
140 :type: Integer Number of samples to return
sagun98
parents:
diff changeset
141 :return List: List of selected samples.
sagun98
parents:
diff changeset
142 :param istmBetaMatrix: File with beta-diversity matrix
sagun98
parents:
diff changeset
143 :type: File stream or file path string
sagun98
parents:
diff changeset
144 """
sagun98
parents:
diff changeset
145
sagun98
parents:
diff changeset
146 #Count of how many rows
sagun98
parents:
diff changeset
147 sampleCount = npaMatrix.shape[0]
sagun98
parents:
diff changeset
148 if iNumberSamplesReturned > sampleCount:
sagun98
parents:
diff changeset
149 logging.error("MicroPITA.funcGetCentralSamplesByKMedoids:: There are not enough samples to return the amount of samples specified. Return sample count = "+str(iNumberSamplesReturned)+". Sample number = "+str(sampleCount)+".")
sagun98
parents:
diff changeset
150 return False
sagun98
parents:
diff changeset
151
sagun98
parents:
diff changeset
152 #If the cluster count is equal to the sample count return all samples
sagun98
parents:
diff changeset
153 if sampleCount == iNumberSamplesReturned:
sagun98
parents:
diff changeset
154 return list(lsSampleNames)
sagun98
parents:
diff changeset
155
sagun98
parents:
diff changeset
156 #Get distance matrix
sagun98
parents:
diff changeset
157 distanceMatrix=scipy.spatial.distance.squareform(Metric.funcReadMatrixFile(istmMatrixFile=istmBetaMatrix,lsSampleOrder=lsSampleNames)[0]) if istmBetaMatrix else Metric.funcGetBetaMetric(npadAbundancies=npaMatrix, sMetric=sMetric, istrmTree=istrmTree, istrmEnvr=istrmEnvr, lsSampleOrder=lsSampleNames)
sagun98
parents:
diff changeset
158 if type(distanceMatrix) is BooleanType:
sagun98
parents:
diff changeset
159 logging.error("MicroPITA.funcGetCentralSamplesByKMedoids:: Could not read in the supplied distance matrix, returning false.")
sagun98
parents:
diff changeset
160 return False
sagun98
parents:
diff changeset
161
sagun98
parents:
diff changeset
162 # Handle unifrac output
sagun98
parents:
diff changeset
163 if sMetric in [Metric.c_strUnifracUnweighted,Metric.c_strUnifracWeighted]:
sagun98
parents:
diff changeset
164 distanceMatrix = distanceMatrix[0]
sagun98
parents:
diff changeset
165
sagun98
parents:
diff changeset
166 #Log distance matrix
sagun98
parents:
diff changeset
167 logging.debug("MicroPITA.funcGetCentralSamplesByKMedoids:: Distance matrix for representative selection using metric="+str(sMetric))
sagun98
parents:
diff changeset
168
sagun98
parents:
diff changeset
169 distance = MLPYDistanceAdaptor(npaDistanceMatrix=distanceMatrix, fIsCondensedMatrix=True)
sagun98
parents:
diff changeset
170
sagun98
parents:
diff changeset
171 #Create object to determine clusters/medoids
sagun98
parents:
diff changeset
172 medoidsMaker = Kmedoids(k=iNumberSamplesReturned, dist=distance)
sagun98
parents:
diff changeset
173 #medoidsData includes(1d numpy array, medoids indexes;
sagun98
parents:
diff changeset
174 # 1d numpy array, non-medoids indexes;
sagun98
parents:
diff changeset
175 # 1d numpy array, cluster membership for non-medoids;
sagun98
parents:
diff changeset
176 # double, cost of configuration)
sagun98
parents:
diff changeset
177 #npaMatrix is samples x rows
sagun98
parents:
diff changeset
178 #Build a matrix of lists of indicies to pass to the distance matrix
sagun98
parents:
diff changeset
179 lliIndicesMatrix = [[iIndexPosition] for iIndexPosition in xrange(0,len(npaMatrix))]
sagun98
parents:
diff changeset
180 medoidsData = medoidsMaker.compute(np.array(lliIndicesMatrix))
sagun98
parents:
diff changeset
181 logging.debug("MicroPITA.funcGetCentralSamplesByKMedoids:: Results from the kmedoid method in representative selection:")
sagun98
parents:
diff changeset
182 logging.debug(str(medoidsData))
sagun98
parents:
diff changeset
183
sagun98
parents:
diff changeset
184 #If returning the same amount of clusters and samples
sagun98
parents:
diff changeset
185 #Return centroids
sagun98
parents:
diff changeset
186 selectedIndexes = medoidsData[0]
sagun98
parents:
diff changeset
187 return [lsSampleNames[selectedIndexes[index]] for index in xrange(0,iNumberSamplesReturned)]
sagun98
parents:
diff changeset
188
sagun98
parents:
diff changeset
189 ####Group 3## Highest Dissimilarity
sagun98
parents:
diff changeset
190 #Testing: Happy path tested
sagun98
parents:
diff changeset
191 def funcSelectExtremeSamplesFromHClust(self, strBetaMetric, npaAbundanceMatrix, lsSampleNames, iSelectSampleCount, istmBetaMatrix=None, istrmTree=None, istrmEnvr=None):
sagun98
parents:
diff changeset
192 """
sagun98
parents:
diff changeset
193 Select extreme samples from HClustering.
sagun98
parents:
diff changeset
194
sagun98
parents:
diff changeset
195 :param strBetaMetric: The beta metric to use for distance matrix generation.
sagun98
parents:
diff changeset
196 :type: String The name of the beta metric to use.
sagun98
parents:
diff changeset
197 :param npaAbundanceMatrix: Numpy array where row=samples and columns=features.
sagun98
parents:
diff changeset
198 :type: Numpy Array Abundance data.
sagun98
parents:
diff changeset
199 :param lsSampleNames: The names of the sample.
sagun98
parents:
diff changeset
200 :type: List List of strings.
sagun98
parents:
diff changeset
201 :param iSelectSampleCount: Number of samples to select (return).
sagun98
parents:
diff changeset
202 :type: Integer Integer number of samples returned.
sagun98
parents:
diff changeset
203 :return Samples: List of samples.
sagun98
parents:
diff changeset
204 :param istmBetaMatrix: File with beta-diversity matrix
sagun98
parents:
diff changeset
205 :type: File stream or file path string
sagun98
parents:
diff changeset
206 """
sagun98
parents:
diff changeset
207
sagun98
parents:
diff changeset
208 #If they want all the sample count, return all sample names
sagun98
parents:
diff changeset
209 iSampleCount=len(npaAbundanceMatrix[:,0])
sagun98
parents:
diff changeset
210 if iSelectSampleCount==iSampleCount:
sagun98
parents:
diff changeset
211 return lsSampleNames
sagun98
parents:
diff changeset
212
sagun98
parents:
diff changeset
213 #Holds the samples to be returned
sagun98
parents:
diff changeset
214 lsReturnSamplesRet = []
sagun98
parents:
diff changeset
215
sagun98
parents:
diff changeset
216 #Generate beta matrix
sagun98
parents:
diff changeset
217 #Returns condensed matrix
sagun98
parents:
diff changeset
218 tempDistanceMatrix = scipy.spatial.distance.squareform(Metric.funcReadMatrixFile(istmMatrixFile=istmBetaMatrix,lsSampleOrder=lsSampleNames)[0]) if istmBetaMatrix else Metric.funcGetBetaMetric(npadAbundancies=npaAbundanceMatrix, sMetric=strBetaMetric, istrmTree=istrmTree, istrmEnvr=istrmEnvr, lsSampleOrder=lsSampleNames, fAdditiveInverse = True)
sagun98
parents:
diff changeset
219
sagun98
parents:
diff changeset
220 if strBetaMetric in [Metric.c_strUnifracUnweighted,Metric.c_strUnifracWeighted]:
sagun98
parents:
diff changeset
221 tempDistanceMatrix = tempDistanceMatrix[0]
sagun98
parents:
diff changeset
222
sagun98
parents:
diff changeset
223 if type(tempDistanceMatrix) is BooleanType:
sagun98
parents:
diff changeset
224 logging.error("MicroPITA.funcSelectExtremeSamplesFromHClust:: Could not read in the supplied distance matrix, returning false.")
sagun98
parents:
diff changeset
225 return False
sagun98
parents:
diff changeset
226
sagun98
parents:
diff changeset
227 if istmBetaMatrix:
sagun98
parents:
diff changeset
228 tempDistanceMatrix = 1-tempDistanceMatrix
sagun98
parents:
diff changeset
229
sagun98
parents:
diff changeset
230 #Feed beta matrix to linkage to cluster
sagun98
parents:
diff changeset
231 #Send condensed matrix
sagun98
parents:
diff changeset
232 linkageMatrix = hcluster.linkage(tempDistanceMatrix, method=self.c_strHierarchicalClusterMethod)
sagun98
parents:
diff changeset
233
sagun98
parents:
diff changeset
234 #Extract cluster information from dendrogram
sagun98
parents:
diff changeset
235 #The linakge matrix is of the form
sagun98
parents:
diff changeset
236 #[[int1 int2 doube int3],...]
sagun98
parents:
diff changeset
237 #int1 and int1 are the paired samples indexed at 0 and up.
sagun98
parents:
diff changeset
238 #each list is an entry for a branch that is number starting with the first
sagun98
parents:
diff changeset
239 #list being sample count index + 1
sagun98
parents:
diff changeset
240 #each list is then named by an increment as they appear
sagun98
parents:
diff changeset
241 #this means that if a number is in the list and is = sample count or greater it is not
sagun98
parents:
diff changeset
242 #terminal and is instead a branch.
sagun98
parents:
diff changeset
243 #This method just takes the lowest metric measurement (highest distance pairs/clusters)
sagun98
parents:
diff changeset
244 #Works much better than the original technique
sagun98
parents:
diff changeset
245 #get total number of samples
sagun98
parents:
diff changeset
246
sagun98
parents:
diff changeset
247 iCurrentSelectCount = 0
sagun98
parents:
diff changeset
248 for row in linkageMatrix:
sagun98
parents:
diff changeset
249 #Get nodes ofthe lowest pairing (so the furthest apart pair)
sagun98
parents:
diff changeset
250 iNode1 = int(row[0])
sagun98
parents:
diff changeset
251 iNode2 = int(row[1])
sagun98
parents:
diff changeset
252 #Make sure the nodes are a terminal node (sample) and not a branch in the dendrogram
sagun98
parents:
diff changeset
253 #The branching in the dendrogram will start at the number of samples and increment higher.
sagun98
parents:
diff changeset
254 #Add each of the pair one at a time breaking when enough samples are selected.
sagun98
parents:
diff changeset
255 if iNode1<iSampleCount:
sagun98
parents:
diff changeset
256 lsReturnSamplesRet.append(lsSampleNames[iNode1])
sagun98
parents:
diff changeset
257 iCurrentSelectCount = iCurrentSelectCount + 1
sagun98
parents:
diff changeset
258 if iCurrentSelectCount == iSelectSampleCount:
sagun98
parents:
diff changeset
259 break
sagun98
parents:
diff changeset
260 if iNode2<iSampleCount:
sagun98
parents:
diff changeset
261 lsReturnSamplesRet.append(lsSampleNames[iNode2])
sagun98
parents:
diff changeset
262 iCurrentSelectCount = iCurrentSelectCount + 1
sagun98
parents:
diff changeset
263 if iCurrentSelectCount == iSelectSampleCount:
sagun98
parents:
diff changeset
264 break
sagun98
parents:
diff changeset
265
sagun98
parents:
diff changeset
266 #Return selected samples
sagun98
parents:
diff changeset
267 return lsReturnSamplesRet
sagun98
parents:
diff changeset
268
sagun98
parents:
diff changeset
269 ####Group 4## Rank Average of user Defined Taxa
sagun98
parents:
diff changeset
270 #Testing: Happy Path Tested
sagun98
parents:
diff changeset
271 def funcGetAverageAbundanceSamples(self, abndTable, lsTargetedFeature, fRank=False):
sagun98
parents:
diff changeset
272 """
sagun98
parents:
diff changeset
273 Averages feature abundance or ranked abundance. Expects a column 0 of taxa id that is skipped.
sagun98
parents:
diff changeset
274
sagun98
parents:
diff changeset
275 :param abndTable: Abundance Table to analyse
sagun98
parents:
diff changeset
276 :type: AbundanceTable Abundance Table
sagun98
parents:
diff changeset
277 :param lsTargetedFeature: String names
sagun98
parents:
diff changeset
278 :type: list list of string names of features (bugs) which are measured after ranking against the full sample
sagun98
parents:
diff changeset
279 :param fRank: Indicates to rank the abundance before getting the average abundance of the features (default false)
sagun98
parents:
diff changeset
280 :type: boolean Flag indicating ranking abundance before calculating average feature measurement (false= no ranking)
sagun98
parents:
diff changeset
281 :return List of lists or boolean: List of lists or False on error. One internal list per sample indicating the sample,
sagun98
parents:
diff changeset
282 feature average abundance or ranked abundance. Lists will already be sorted.
sagun98
parents:
diff changeset
283 For not Ranked [[sample,average abundance of selected feature,1]]
sagun98
parents:
diff changeset
284 For Ranked [[sample,average ranked abundance, average abundance of selected feature]]
sagun98
parents:
diff changeset
285 Error Returns false
sagun98
parents:
diff changeset
286 """
sagun98
parents:
diff changeset
287
sagun98
parents:
diff changeset
288 llAbundance = abndTable.funcGetAverageAbundancePerSample(lsTargetedFeature)
sagun98
parents:
diff changeset
289 if not llAbundance:
sagun98
parents:
diff changeset
290 logging.error("MicroPITA.funcGetAverageAbundanceSamples:: Could not get average abundance, returned false. Make sure the features (bugs) are spelled correctly and in the abundance table.")
sagun98
parents:
diff changeset
291 return False
sagun98
parents:
diff changeset
292 #Add a space for ranking if needed
sagun98
parents:
diff changeset
293 #Not ranked will be [[sSample,average abundance,1]]
sagun98
parents:
diff changeset
294 #(where 1 will not discriminant ties if used in later functions, so this generalizes)
sagun98
parents:
diff changeset
295 #Ranked will be [[sSample, average rank, average abundance]]
sagun98
parents:
diff changeset
296 llRetAbundance = [[llist[0],-1,llist[1]] for llist in llAbundance]
sagun98
parents:
diff changeset
297 #Rank if needed
sagun98
parents:
diff changeset
298 if fRank:
sagun98
parents:
diff changeset
299 abndRanked = abndTable.funcRankAbundance()
sagun98
parents:
diff changeset
300 if abndRanked == None:
sagun98
parents:
diff changeset
301 logging.error("MicroPITA.funcGetAverageAbundanceSamples:: Could not rank the abundance table, returned false.")
sagun98
parents:
diff changeset
302 return False
sagun98
parents:
diff changeset
303 llRetRank = abndRanked.funcGetAverageAbundancePerSample(lsTargetedFeature)
sagun98
parents:
diff changeset
304 if not llRetRank:
sagun98
parents:
diff changeset
305 logging.error("MicroPITA.funcGetAverageAbundanceSamples:: Could not get average ranked abundance, returned false. Make sure the features (bugs) are spelled correctly and in the abundance table.")
sagun98
parents:
diff changeset
306 return False
sagun98
parents:
diff changeset
307 dictRanks = dict(llRetRank)
sagun98
parents:
diff changeset
308 llRetAbundance = [[a[0],dictRanks[a[0]],a[2]] for a in llRetAbundance]
sagun98
parents:
diff changeset
309
sagun98
parents:
diff changeset
310 #Sort first for ties and then for the main feature
sagun98
parents:
diff changeset
311 if not fRank or ConstantsMicropita.c_fBreakRankTiesByDiversity:
sagun98
parents:
diff changeset
312 llRetAbundance = sorted(llRetAbundance, key = lambda sampleData: sampleData[2], reverse = not fRank)
sagun98
parents:
diff changeset
313 if fRank:
sagun98
parents:
diff changeset
314 llRetAbundance = sorted(llRetAbundance, key = lambda sampleData: sampleData[1], reverse = not fRank)
sagun98
parents:
diff changeset
315 return llRetAbundance
sagun98
parents:
diff changeset
316
sagun98
parents:
diff changeset
317 #Testing: Happy Path Tested
sagun98
parents:
diff changeset
318 def funcSelectTargetedTaxaSamples(self, abndMatrix, lsTargetedTaxa, iSampleSelectionCount, sMethod = ConstantsMicropita.lsTargetedFeatureMethodValues[0]):
sagun98
parents:
diff changeset
319 """
sagun98
parents:
diff changeset
320 Selects samples with the highest ranks or abundance of targeted features.
sagun98
parents:
diff changeset
321 If ranked, select the highest abundance for tie breaking
sagun98
parents:
diff changeset
322
sagun98
parents:
diff changeset
323 :param abndMatrix: Abundance table to analyse
sagun98
parents:
diff changeset
324 :type: AbundanceTable Abundance table
sagun98
parents:
diff changeset
325 :param lsTargetedTaxa: List of features
sagun98
parents:
diff changeset
326 :type: list list of strings
sagun98
parents:
diff changeset
327 :param iSampleSelectionCount: Number of samples to select
sagun98
parents:
diff changeset
328 :type: integer integer
sagun98
parents:
diff changeset
329 :param sMethod: Method to select targeted features
sagun98
parents:
diff changeset
330 :type: string String (Can be values found in ConstantsMicropita.lsTargetedFeatureMethodValues)
sagun98
parents:
diff changeset
331 :return List of strings: List of sample names which were selected
sagun98
parents:
diff changeset
332 List of strings Empty list is returned on an error.
sagun98
parents:
diff changeset
333 """
sagun98
parents:
diff changeset
334
sagun98
parents:
diff changeset
335 #Check data
sagun98
parents:
diff changeset
336 if(len(lsTargetedTaxa) < 1):
sagun98
parents:
diff changeset
337 logging.error("MicroPITA.funcSelectTargetedTaxaSamples. Taxa defined selection was requested but no features were given.")
sagun98
parents:
diff changeset
338 return []
sagun98
parents:
diff changeset
339
sagun98
parents:
diff changeset
340 lsTargetedSamples = self.funcGetAverageAbundanceSamples(abndTable=abndMatrix, lsTargetedFeature=lsTargetedTaxa,
sagun98
parents:
diff changeset
341 fRank=sMethod.lower() == self.c_strTargetedRanked.lower())
sagun98
parents:
diff changeset
342 #If an error occured or the key word for the method was not recognized
sagun98
parents:
diff changeset
343 if lsTargetedSamples == False:
sagun98
parents:
diff changeset
344 logging.error("MicroPITA.funcSelectTargetedTaxaSamples:: Was not able to select for the features given. So targeted feature selection was performed. Check to make sure the features are spelled correctly and exist in the abundance file.")
sagun98
parents:
diff changeset
345 return []
sagun98
parents:
diff changeset
346
sagun98
parents:
diff changeset
347 #Select from results
sagun98
parents:
diff changeset
348 return [sSample[0] for sSample in lsTargetedSamples[:iSampleSelectionCount]]
sagun98
parents:
diff changeset
349
sagun98
parents:
diff changeset
350 ####Group 5## Random
sagun98
parents:
diff changeset
351 #Testing: Happy path Tested
sagun98
parents:
diff changeset
352 def funcGetRandomSamples(self, lsSamples=None, iNumberOfSamplesToReturn=0):
sagun98
parents:
diff changeset
353 """
sagun98
parents:
diff changeset
354 Returns random sample names of the number given. No replacement.
sagun98
parents:
diff changeset
355
sagun98
parents:
diff changeset
356 :param lsSamples: List of sample names
sagun98
parents:
diff changeset
357 :type: list list of strings
sagun98
parents:
diff changeset
358 :param iNumberOfSamplesToReturn: Number of samples to select
sagun98
parents:
diff changeset
359 :type: integer integer.
sagun98
parents:
diff changeset
360 :return List: List of selected samples (strings).
sagun98
parents:
diff changeset
361 """
sagun98
parents:
diff changeset
362
sagun98
parents:
diff changeset
363 #Input matrix sample count
sagun98
parents:
diff changeset
364 sampleCount = len(lsSamples)
sagun98
parents:
diff changeset
365
sagun98
parents:
diff changeset
366 #Return the full matrix if they ask for a return matrix where length == original
sagun98
parents:
diff changeset
367 if(iNumberOfSamplesToReturn >= sampleCount):
sagun98
parents:
diff changeset
368 return lsSamples
sagun98
parents:
diff changeset
369
sagun98
parents:
diff changeset
370 #Get the random indices for the sample (without replacement)
sagun98
parents:
diff changeset
371 liRandomIndices = random.sample(range(sampleCount), iNumberOfSamplesToReturn)
sagun98
parents:
diff changeset
372
sagun98
parents:
diff changeset
373 #Create a boolean array of if indexes are to be included in the reduced array
sagun98
parents:
diff changeset
374 return [sSample for iIndex, sSample in enumerate(lsSamples) if iIndex in liRandomIndices]
sagun98
parents:
diff changeset
375
sagun98
parents:
diff changeset
376 #Happy path tested (case 3)
sagun98
parents:
diff changeset
377 def funcGetAveragePopulation(self, abndTable, lfCompress):
sagun98
parents:
diff changeset
378 """
sagun98
parents:
diff changeset
379 Get the average row per column in the abndtable.
sagun98
parents:
diff changeset
380
sagun98
parents:
diff changeset
381 :param abndTable: AbundanceTable of data to be averaged
sagun98
parents:
diff changeset
382 :type: AbudanceTable
sagun98
parents:
diff changeset
383 :param lfCompress: List of boolean flags (false means to remove sample before averaging
sagun98
parents:
diff changeset
384 :type: List of floats
sagun98
parents:
diff changeset
385 :return List of doubles:
sagun98
parents:
diff changeset
386 """
sagun98
parents:
diff changeset
387 if sum(lfCompress) == 0:
sagun98
parents:
diff changeset
388 return []
sagun98
parents:
diff changeset
389
sagun98
parents:
diff changeset
390 #Get the average populations
sagun98
parents:
diff changeset
391 lAverageRet = []
sagun98
parents:
diff changeset
392
sagun98
parents:
diff changeset
393 for sFeature in abndTable.funcGetAbundanceCopy():
sagun98
parents:
diff changeset
394 sFeature = list(sFeature)[1:]
sagun98
parents:
diff changeset
395 sFeature=np.compress(lfCompress,sFeature,axis=0)
sagun98
parents:
diff changeset
396 lAverageRet.append(sum(sFeature)/float(len(sFeature)))
sagun98
parents:
diff changeset
397 return lAverageRet
sagun98
parents:
diff changeset
398
sagun98
parents:
diff changeset
399 #Happy path tested (2 cases)
sagun98
parents:
diff changeset
400 def funcGetDistanceFromAverage(self, abndTable,ldAverage,lsSamples,lfSelected):
sagun98
parents:
diff changeset
401 """
sagun98
parents:
diff changeset
402 Given an abundance table and an average sample, this returns the distance of each sample
sagun98
parents:
diff changeset
403 (measured using brays-curtis dissimilarity) from the average.
sagun98
parents:
diff changeset
404 The distances are reduced by needing to be in the lsSamples and being a true in the lfSelected
sagun98
parents:
diff changeset
405 (which is associated with the samples in the order of the samples in the abundance table;
sagun98
parents:
diff changeset
406 use abundancetable.funcGetSampleNames() to see the order if needed).
sagun98
parents:
diff changeset
407
sagun98
parents:
diff changeset
408 :param abndTable: Abundance table holding the data to be analyzed.
sagun98
parents:
diff changeset
409 :type: AbundanceTable
sagun98
parents:
diff changeset
410 :param ldAverage: Average population (Average features of the abundance table of samples)
sagun98
parents:
diff changeset
411 :type: List of doubles which represent the average population
sagun98
parents:
diff changeset
412 :param lsSamples: These are the only samples used in the analysis
sagun98
parents:
diff changeset
413 :type: List of strings (sample ids)
sagun98
parents:
diff changeset
414 :param lfSelected: Samples to be included in the analysis
sagun98
parents:
diff changeset
415 :type: List of boolean (true means include)
sagun98
parents:
diff changeset
416 :return: List of distances (doubles)
sagun98
parents:
diff changeset
417 """
sagun98
parents:
diff changeset
418 #Get the distance from label 1 of all samples in label0 splitting into selected and not selected lists
sagun98
parents:
diff changeset
419 ldSelectedDistances = []
sagun98
parents:
diff changeset
420
sagun98
parents:
diff changeset
421 for sSampleName in [sSample for iindex, sSample in enumerate(lsSamples) if lfSelected[iindex]]:
sagun98
parents:
diff changeset
422 #Get the sample measurements
sagun98
parents:
diff changeset
423 ldSelectedDistances.append(Metric.funcGetBrayCurtisDissimilarity(np.array([abndTable.funcGetSample(sSampleName),ldAverage]))[0])
sagun98
parents:
diff changeset
424 return ldSelectedDistances
sagun98
parents:
diff changeset
425
sagun98
parents:
diff changeset
426 #Happy path tested (1 case)
sagun98
parents:
diff changeset
427 def funcMeasureDistanceFromLabelToAverageOtherLabel(self, abndTable, lfGroupOfInterest, lfGroupOther):
sagun98
parents:
diff changeset
428 """
sagun98
parents:
diff changeset
429 Get the distance of samples from one label from the average sample of not the label.
sagun98
parents:
diff changeset
430 Note: This assumes 2 classes.
sagun98
parents:
diff changeset
431
sagun98
parents:
diff changeset
432 :param abndTable: Table of data to work out of.
sagun98
parents:
diff changeset
433 :type: Abundace Table
sagun98
parents:
diff changeset
434 :param lfGroupOfInterest: Boolean indicator of the sample being in the first group.
sagun98
parents:
diff changeset
435 :type: List of floats, true indicating an individual in the group of interest.
sagun98
parents:
diff changeset
436 :param lfGroupOther: Boolean indicator of the sample being in the other group.
sagun98
parents:
diff changeset
437 :type: List of floats, true indicating an individual in the
sagun98
parents:
diff changeset
438 :return List of List of doubles: [list of tuples (string sample name,double distance) for the selected population, list of tuples for the not selected population]
sagun98
parents:
diff changeset
439 """
sagun98
parents:
diff changeset
440 #Get all sample names
sagun98
parents:
diff changeset
441 lsAllSamples = abndTable.funcGetSampleNames()
sagun98
parents:
diff changeset
442
sagun98
parents:
diff changeset
443 #Get average populations
sagun98
parents:
diff changeset
444 lAverageOther = self.funcGetAveragePopulation(abndTable=abndTable, lfCompress=lfGroupOther)
sagun98
parents:
diff changeset
445
sagun98
parents:
diff changeset
446 #Get the distance from the average of the other label (label 1)
sagun98
parents:
diff changeset
447 ldSelectedDistances = self.funcGetDistanceFromAverage(abndTable=abndTable, ldAverage=lAverageOther,
sagun98
parents:
diff changeset
448 lsSamples=lsAllSamples, lfSelected=lfGroupOfInterest)
sagun98
parents:
diff changeset
449
sagun98
parents:
diff changeset
450 return zip([lsAllSamples[iindex] for iindex, fGroup in enumerate(lfGroupOfInterest) if fGroup],ldSelectedDistances)
sagun98
parents:
diff changeset
451
sagun98
parents:
diff changeset
452 #Happy path tested (1 test case)
sagun98
parents:
diff changeset
453 def funcPerformDistanceSelection(self, abndTable, iSelectionCount, sLabel, sValueOfInterest):
sagun98
parents:
diff changeset
454 """
sagun98
parents:
diff changeset
455 Given metadata, metadata of one value (sValueOfInterest) is measured from the average (centroid) value of another label group.
sagun98
parents:
diff changeset
456 An iSelectionCount of samples is selected from the group of interest closest to and furthest from the centroid of the other group.
sagun98
parents:
diff changeset
457
sagun98
parents:
diff changeset
458 :params abndTable: Abundance of measurements
sagun98
parents:
diff changeset
459 :type: AbundanceTable
sagun98
parents:
diff changeset
460 :params iSelectionCount: The number of samples selected per sample.
sagun98
parents:
diff changeset
461 :type: Integer Integer greater than 0
sagun98
parents:
diff changeset
462 :params sLabel: ID of the metadata which is the supervised label
sagun98
parents:
diff changeset
463 :type: String
sagun98
parents:
diff changeset
464 :params sValueOfInterest: Metadata value in the sLabel metadta row of the abundance table which defines the group of interest.
sagun98
parents:
diff changeset
465 :type: String found in the abundance table metadata row indicated by sLabel.
sagun98
parents:
diff changeset
466 :return list list of tuples (samplename, distance) [[iSelectionCount of tuples closest to the other centroid], [iSelectionCount of tuples farthest from the other centroid], [all tuples of samples not selected]]
sagun98
parents:
diff changeset
467 """
sagun98
parents:
diff changeset
468
sagun98
parents:
diff changeset
469 lsMetadata = abndTable.funcGetMetadata(sLabel)
sagun98
parents:
diff changeset
470 #Other metadata values
sagun98
parents:
diff changeset
471 lsUniqueOtherValues = list(set(lsMetadata)-set(sValueOfInterest))
sagun98
parents:
diff changeset
472
sagun98
parents:
diff changeset
473 #Get boolean indicator of values of interest
sagun98
parents:
diff changeset
474 lfLabelsInterested = [sValueOfInterest == sValue for sValue in lsMetadata]
sagun98
parents:
diff changeset
475
sagun98
parents:
diff changeset
476 #Get the distances of the items of interest from the other metadata values
sagun98
parents:
diff changeset
477 dictDistanceAverages = {}
sagun98
parents:
diff changeset
478 for sOtherLabel in lsUniqueOtherValues:
sagun98
parents:
diff changeset
479 #Get boolean indicator of labels not of interest
sagun98
parents:
diff changeset
480 lfLabelsOther = [sOtherLabel == sValue for sValue in lsMetadata]
sagun98
parents:
diff changeset
481
sagun98
parents:
diff changeset
482 #Get the distances of data from two different groups to the average of the other
sagun98
parents:
diff changeset
483 ldValueDistances = dict(self.funcMeasureDistanceFromLabelToAverageOtherLabel(abndTable, lfLabelsInterested, lfLabelsOther))
sagun98
parents:
diff changeset
484
sagun98
parents:
diff changeset
485 for sKey in ldValueDistances:
sagun98
parents:
diff changeset
486 dictDistanceAverages[sKey] = ldValueDistances[sKey] + dictDistanceAverages[sKey] if sKey in dictDistanceAverages else ldValueDistances[sKey]
sagun98
parents:
diff changeset
487
sagun98
parents:
diff changeset
488 #Finish average by dividing by length of lsUniqueOtherValues
sagun98
parents:
diff changeset
489 ltpleAverageDistances = [(sKey, dictDistanceAverages[sKey]/float(len(lsUniqueOtherValues))) for sKey in dictDistanceAverages]
sagun98
parents:
diff changeset
490
sagun98
parents:
diff changeset
491 #Sort to extract extremes
sagun98
parents:
diff changeset
492 ltpleAverageDistances = sorted(ltpleAverageDistances,key=operator.itemgetter(1))
sagun98
parents:
diff changeset
493
sagun98
parents:
diff changeset
494 #Get the closest and farthest distances
sagun98
parents:
diff changeset
495 ltupleDiscriminantSamples = ltpleAverageDistances[:iSelectionCount]
sagun98
parents:
diff changeset
496 ltupleDistinctSamples = ltpleAverageDistances[iSelectionCount*-1:]
sagun98
parents:
diff changeset
497
sagun98
parents:
diff changeset
498 #Remove the selected samples from the larger population of distances (better visualization)
sagun98
parents:
diff changeset
499 ldSelected = [tpleSelected[0] for tpleSelected in ltupleDiscriminantSamples+ltupleDistinctSamples]
sagun98
parents:
diff changeset
500
sagun98
parents:
diff changeset
501 #Return discriminant tuples, distinct tuples, other tuples
sagun98
parents:
diff changeset
502 return [ltupleDiscriminantSamples, ltupleDistinctSamples,
sagun98
parents:
diff changeset
503 [tplData for tplData in ltpleAverageDistances if tplData[0] not in ldSelected]]
sagun98
parents:
diff changeset
504
sagun98
parents:
diff changeset
505 #Run the supervised method surrounding distance from centroids
sagun98
parents:
diff changeset
506 #Happy path tested (3 test cases)
sagun98
parents:
diff changeset
507 def funcRunSupervisedDistancesFromCentroids(self, abundanceTable, fRunDistinct, fRunDiscriminant,
sagun98
parents:
diff changeset
508 xOutputSupFile, xPredictSupFile, strSupervisedMetadata,
sagun98
parents:
diff changeset
509 iSampleSupSelectionCount, lsOriginalSampleNames, lsOriginalLabels, fAppendFiles = False):
sagun98
parents:
diff changeset
510 """
sagun98
parents:
diff changeset
511 Runs supervised methods based on measuring distances of one label from the centroid of another. NAs are evaluated as theirown group.
sagun98
parents:
diff changeset
512
sagun98
parents:
diff changeset
513 :param abundanceTable: AbundanceTable
sagun98
parents:
diff changeset
514 :type: AbudanceTable Data to analyze
sagun98
parents:
diff changeset
515 :param fRunDistinct: Run distinct selection method
sagun98
parents:
diff changeset
516 :type: Boolean boolean (true runs method)
sagun98
parents:
diff changeset
517 :param fRunDiscriminant: Run discriminant method
sagun98
parents:
diff changeset
518 :type: Boolean boolean (true runs method)
sagun98
parents:
diff changeset
519 :param xOutputSupFile: File output from supervised methods detailing data going into the method.
sagun98
parents:
diff changeset
520 :type: String or FileStream
sagun98
parents:
diff changeset
521 :param xPredictSupFile: File output from supervised methods distance results from supervised methods.
sagun98
parents:
diff changeset
522 :type: String or FileStream
sagun98
parents:
diff changeset
523 :param strSupervisedMetadata: The metadata that will be used to group samples.
sagun98
parents:
diff changeset
524 :type: String
sagun98
parents:
diff changeset
525 :param iSampleSupSelectionCount: Number of samples to select
sagun98
parents:
diff changeset
526 :type: Integer int sample selection count
sagun98
parents:
diff changeset
527 :param lsOriginalSampleNames: List of the sample names, order is important and should be preserved from the abundanceTable.
sagun98
parents:
diff changeset
528 :type: List of samples
sagun98
parents:
diff changeset
529 :param fAppendFiles: Indicates that output files already exist and appending is occuring.
sagun98
parents:
diff changeset
530 :type: Boolean
sagun98
parents:
diff changeset
531 :return Selected Samples: A dictionary of selected samples by selection ID
sagun98
parents:
diff changeset
532 Dictionary {"Selection Method":["SampleID","SampleID"...]}
sagun98
parents:
diff changeset
533 """
sagun98
parents:
diff changeset
534 #Get labels and run one label against many
sagun98
parents:
diff changeset
535 lstrMetadata = abundanceTable.funcGetMetadata(strSupervisedMetadata)
sagun98
parents:
diff changeset
536 dictlltpleDistanceMeasurements = {}
sagun98
parents:
diff changeset
537 for sMetadataValue in set(lstrMetadata):
sagun98
parents:
diff changeset
538
sagun98
parents:
diff changeset
539 #For now perform the selection here for the label of interest against the other labels
sagun98
parents:
diff changeset
540 dictlltpleDistanceMeasurements.setdefault(sMetadataValue,[]).extend(self.funcPerformDistanceSelection(abndTable=abundanceTable,
sagun98
parents:
diff changeset
541 iSelectionCount=iSampleSupSelectionCount, sLabel=strSupervisedMetadata, sValueOfInterest=sMetadataValue))
sagun98
parents:
diff changeset
542
sagun98
parents:
diff changeset
543 #Make expected output files for supervised methods
sagun98
parents:
diff changeset
544 #1. Output file which is similar to an input file for SVMs
sagun98
parents:
diff changeset
545 #2. Output file that is similar to the probabilitic output of a SVM (LibSVM)
sagun98
parents:
diff changeset
546 #Manly for making output of supervised methods (Distance from Centroid) similar
sagun98
parents:
diff changeset
547 #MicropitaVis needs some of these files
sagun98
parents:
diff changeset
548 if xOutputSupFile:
sagun98
parents:
diff changeset
549 if fAppendFiles:
sagun98
parents:
diff changeset
550 SVM.funcUpdateSVMFileWithAbundanceTable(abndAbundanceTable=abundanceTable, xOutputSVMFile=xOutputSupFile,
sagun98
parents:
diff changeset
551 lsOriginalLabels=lsOriginalLabels, lsSampleOrdering=lsOriginalSampleNames)
sagun98
parents:
diff changeset
552 else:
sagun98
parents:
diff changeset
553 SVM.funcConvertAbundanceTableToSVMFile(abndAbundanceTable=abundanceTable, xOutputSVMFile=xOutputSupFile,
sagun98
parents:
diff changeset
554 sMetadataLabel=strSupervisedMetadata, lsOriginalLabels=lsOriginalLabels, lsSampleOrdering=lsOriginalSampleNames)
sagun98
parents:
diff changeset
555
sagun98
parents:
diff changeset
556 #Will contain the samples selected to return
sagun98
parents:
diff changeset
557 #One or more of the methods may be active so this is why I am extending instead of just returning the result of each method type
sagun98
parents:
diff changeset
558 dictSelectedSamplesRet = dict()
sagun98
parents:
diff changeset
559 for sKey, ltplDistances in dictlltpleDistanceMeasurements.items():
sagun98
parents:
diff changeset
560 if fRunDistinct:
sagun98
parents:
diff changeset
561 dictSelectedSamplesRet.setdefault(ConstantsMicropita.c_strDistinct,[]).extend([ltple[0] for ltple in ltplDistances[1]])
sagun98
parents:
diff changeset
562 if fRunDiscriminant:
sagun98
parents:
diff changeset
563 dictSelectedSamplesRet.setdefault(ConstantsMicropita.c_strDiscriminant,[]).extend([ltple[0] for ltple in ltplDistances[0]])
sagun98
parents:
diff changeset
564
sagun98
parents:
diff changeset
565 if xPredictSupFile:
sagun98
parents:
diff changeset
566 dictFlattenedDistances = dict()
sagun98
parents:
diff changeset
567 [dictFlattenedDistances.setdefault(sKey, []).append(tple)
sagun98
parents:
diff changeset
568 for sKey, lltple in dictlltpleDistanceMeasurements.items()
sagun98
parents:
diff changeset
569 for ltple in lltple for tple in ltple]
sagun98
parents:
diff changeset
570 if fAppendFiles:
sagun98
parents:
diff changeset
571 self._updatePredictFile(xPredictSupFile=xPredictSupFile, xInputLabelsFile=xOutputSupFile,
sagun98
parents:
diff changeset
572 dictltpleDistanceMeasurements=dictFlattenedDistances, abundanceTable=abundanceTable, lsOriginalSampleNames=lsOriginalSampleNames)
sagun98
parents:
diff changeset
573 else:
sagun98
parents:
diff changeset
574 self._writeToPredictFile(xPredictSupFile=xPredictSupFile, xInputLabelsFile=xOutputSupFile,
sagun98
parents:
diff changeset
575 dictltpleDistanceMeasurements=dictFlattenedDistances, abundanceTable=abundanceTable, lsOriginalSampleNames=lsOriginalSampleNames)
sagun98
parents:
diff changeset
576 return dictSelectedSamplesRet
sagun98
parents:
diff changeset
577
sagun98
parents:
diff changeset
578 #Two happy path test cases
sagun98
parents:
diff changeset
579 def _updatePredictFile(self, xPredictSupFile, xInputLabelsFile, dictltpleDistanceMeasurements, abundanceTable, lsOriginalSampleNames):
sagun98
parents:
diff changeset
580 """
sagun98
parents:
diff changeset
581 Manages updating the predict file.
sagun98
parents:
diff changeset
582
sagun98
parents:
diff changeset
583 :param xPredictSupFile: File that has predictions (distances) from the supervised method.
sagun98
parents:
diff changeset
584 :type: FileStream or String file path
sagun98
parents:
diff changeset
585 :param xInputLabelsFile: File that as input to the supervised methods.
sagun98
parents:
diff changeset
586 :type: FileStream or String file path
sagun98
parents:
diff changeset
587 :param dictltpleDistanceMeasurements:
sagun98
parents:
diff changeset
588 :type: Dictionary of lists of tuples {"labelgroup":[("SampleName",dDistance)...], "labelgroup":[("SampleName",dDistance)...]}
sagun98
parents:
diff changeset
589 """
sagun98
parents:
diff changeset
590
sagun98
parents:
diff changeset
591 if not isinstance(xPredictSupFile, str):
sagun98
parents:
diff changeset
592 xPredictSupFile.close()
sagun98
parents:
diff changeset
593 xPredictSupFile = xPredictSupFile.name
sagun98
parents:
diff changeset
594 csvr = open(xPredictSupFile,'r')
sagun98
parents:
diff changeset
595
sagun98
parents:
diff changeset
596 f = csv.reader(csvr,delimiter=ConstantsBreadCrumbs.c_strBreadCrumbsSVMSpace)
sagun98
parents:
diff changeset
597 lsHeader = f.next()[1:]
sagun98
parents:
diff changeset
598 dictlltpleRead = dict([(sHeader,[]) for sHeader in lsHeader])
sagun98
parents:
diff changeset
599
sagun98
parents:
diff changeset
600 #Read data in
sagun98
parents:
diff changeset
601 iSampleIndex = 0
sagun98
parents:
diff changeset
602 for sRow in f:
sagun98
parents:
diff changeset
603 sLabel = sRow[0]
sagun98
parents:
diff changeset
604 [dictlltpleRead[lsHeader[iDistanceIndex]].append((lsOriginalSampleNames[iSampleIndex],dDistance)) for iDistanceIndex, dDistance in enumerate(sRow[1:])
sagun98
parents:
diff changeset
605 if not dDistance == ConstantsMicropita.c_sEmptyPredictFileValue]
sagun98
parents:
diff changeset
606 iSampleIndex += 1
sagun98
parents:
diff changeset
607
sagun98
parents:
diff changeset
608 #Combine dictltpleDistanceMeasurements with new data
sagun98
parents:
diff changeset
609 #If they share a key then merge keeping parameter data
sagun98
parents:
diff changeset
610 #If they do not share the key, keep the full data
sagun98
parents:
diff changeset
611 dictNew = {}
sagun98
parents:
diff changeset
612 for sKey in dictltpleDistanceMeasurements.keys():
sagun98
parents:
diff changeset
613 lsSamples = [tple[0] for tple in dictltpleDistanceMeasurements[sKey]]
sagun98
parents:
diff changeset
614 dictNew[sKey] = dictltpleDistanceMeasurements[sKey]+[tple for tple in dictlltpleRead[sKey] if tple[0] not in lsSamples] if sKey in dictlltpleRead.keys() else dictltpleDistanceMeasurements[sKey]
sagun98
parents:
diff changeset
615 for sKey in dictlltpleRead:
sagun98
parents:
diff changeset
616 if sKey not in dictltpleDistanceMeasurements.keys():
sagun98
parents:
diff changeset
617 dictNew[sKey] = dictlltpleRead[sKey]
sagun98
parents:
diff changeset
618
sagun98
parents:
diff changeset
619 #Call writer
sagun98
parents:
diff changeset
620 self._writeToPredictFile(xPredictSupFile=xPredictSupFile, xInputLabelsFile=xInputLabelsFile,
sagun98
parents:
diff changeset
621 dictltpleDistanceMeasurements=dictNew, abundanceTable=abundanceTable,
sagun98
parents:
diff changeset
622 lsOriginalSampleNames=lsOriginalSampleNames, fFromUpdate=True)
sagun98
parents:
diff changeset
623
sagun98
parents:
diff changeset
624 #2 happy path test cases
sagun98
parents:
diff changeset
625 def _writeToPredictFile(self, xPredictSupFile, xInputLabelsFile, dictltpleDistanceMeasurements, abundanceTable, lsOriginalSampleNames, fFromUpdate=False):
sagun98
parents:
diff changeset
626 """
sagun98
parents:
diff changeset
627 Write to the predict file.
sagun98
parents:
diff changeset
628
sagun98
parents:
diff changeset
629 :param xPredictSupFile: File that has predictions (distances) from the supervised method.
sagun98
parents:
diff changeset
630 :type: FileStream or String file path
sagun98
parents:
diff changeset
631 :param xInputLabelsFile: File that as input to the supervised methods.
sagun98
parents:
diff changeset
632 :type: FileStream or String file path
sagun98
parents:
diff changeset
633 :param dictltpleDistanceMeasurements:
sagun98
parents:
diff changeset
634 :type: Dictionary of lists of tuples {"labelgroup":[("SampleName",dDistance)...], "labelgroup":[("SampleName",dDistance)...]}
sagun98
parents:
diff changeset
635 :param abundanceTable: An abundance table of the sample data.
sagun98
parents:
diff changeset
636 :type: AbundanceTable
sagun98
parents:
diff changeset
637 :param lsOriginalSampleNames: Used if the file is being updated as the sample names so that it may be passed in and consistent with other writing.
sagun98
parents:
diff changeset
638 Otherwise will use the sample names from the abundance table.
sagun98
parents:
diff changeset
639 :type: List of strings
sagun98
parents:
diff changeset
640 :param fFromUpdate: Indicates if this is part of an update to the file or not.
sagun98
parents:
diff changeset
641 :type: Boolean
sagun98
parents:
diff changeset
642 """
sagun98
parents:
diff changeset
643
sagun98
parents:
diff changeset
644 xInputLabelsFileName = xInputLabelsFile
sagun98
parents:
diff changeset
645 if not isinstance(xInputLabelsFile,str):
sagun98
parents:
diff changeset
646 xInputLabelsFileName = xInputLabelsFile.name
sagun98
parents:
diff changeset
647 f = csv.writer(open(xPredictSupFile,"w") if isinstance(xPredictSupFile, str) else xPredictSupFile,delimiter=ConstantsBreadCrumbs.c_strBreadCrumbsSVMSpace)
sagun98
parents:
diff changeset
648
sagun98
parents:
diff changeset
649 lsAllSampleNames = abundanceTable.funcGetSampleNames()
sagun98
parents:
diff changeset
650 lsLabels = SVM.funcReadLabelsFromFile(xSVMFile=xInputLabelsFileName, lsAllSampleNames= lsOriginalSampleNames if fFromUpdate else lsAllSampleNames,
sagun98
parents:
diff changeset
651 isPredictFile=False)
sagun98
parents:
diff changeset
652 dictLabels = dict([(sSample,sLabel) for sLabel in lsLabels.keys() for sSample in lsLabels[sLabel]])
sagun98
parents:
diff changeset
653
sagun98
parents:
diff changeset
654 #Dictionay keys will be used to order the predict file
sagun98
parents:
diff changeset
655 lsMeasurementKeys = dictltpleDistanceMeasurements.keys()
sagun98
parents:
diff changeset
656 #Make header
sagun98
parents:
diff changeset
657 f.writerow(["labels"]+lsMeasurementKeys)
sagun98
parents:
diff changeset
658
sagun98
parents:
diff changeset
659 #Reformat dictionary to make it easier to use
sagun98
parents:
diff changeset
660 for sKey in dictltpleDistanceMeasurements:
sagun98
parents:
diff changeset
661 dictltpleDistanceMeasurements[sKey] = dict([ltpl for ltpl in dictltpleDistanceMeasurements[sKey]])
sagun98
parents:
diff changeset
662
sagun98
parents:
diff changeset
663 for sSample in lsOriginalSampleNames:
sagun98
parents:
diff changeset
664 #Make body of file
sagun98
parents:
diff changeset
665 f.writerow([dictLabels.get(sSample,ConstantsMicropita.c_sEmptyPredictFileValue)]+
sagun98
parents:
diff changeset
666 [str(dictltpleDistanceMeasurements[sKey].get(sSample,ConstantsMicropita.c_sEmptyPredictFileValue))
sagun98
parents:
diff changeset
667 for sKey in lsMeasurementKeys])
sagun98
parents:
diff changeset
668
sagun98
parents:
diff changeset
669 def _funcRunNormalizeSensitiveMethods(self, abndData, iSampleSelectionCount, dictSelectedSamples, lsAlphaMetrics, lsBetaMetrics, lsInverseBetaMetrics,
sagun98
parents:
diff changeset
670 fRunDiversity, fRunRepresentative, fRunExtreme, strAlphaMetadata=None,
sagun98
parents:
diff changeset
671 istmBetaMatrix=None, istrmTree=None, istrmEnvr=None, fInvertDiversity=False):
sagun98
parents:
diff changeset
672 """
sagun98
parents:
diff changeset
673 Manages running methods that are sensitive to normalization. This is called twice, once for the set of methods which should not be normalized and the other
sagun98
parents:
diff changeset
674 for the set that should be normalized.
sagun98
parents:
diff changeset
675
sagun98
parents:
diff changeset
676 :param abndData: Abundance table object holding the samples to be measured.
sagun98
parents:
diff changeset
677 :type: AbundanceTable
sagun98
parents:
diff changeset
678 :param iSampleSelectionCount The number of samples to select per method.
sagun98
parents:
diff changeset
679 :type: Integer
sagun98
parents:
diff changeset
680 :param dictSelectedSamples Will be added to as samples are selected {"Method:["strSelectedSampleID","strSelectedSampleID"...]}.
sagun98
parents:
diff changeset
681 :type: Dictionary
sagun98
parents:
diff changeset
682 :param lsAlphaMetrics: List of alpha metrics to use on alpha metric dependent assays (like highest diversity).
sagun98
parents:
diff changeset
683 :type: List of strings
sagun98
parents:
diff changeset
684 :param lsBetaMetrics: List of beta metrics to use on beta metric dependent assays (like most representative).
sagun98
parents:
diff changeset
685 :type: List of strings
sagun98
parents:
diff changeset
686 :param lsInverseBetaMetrics: List of inverse beta metrics to use on inverse beta metric dependent assays (like most dissimilar).
sagun98
parents:
diff changeset
687 :type: List of strings
sagun98
parents:
diff changeset
688 :param fRunDiversity: Run Diversity based methods (true indicates run).
sagun98
parents:
diff changeset
689 :type: Boolean
sagun98
parents:
diff changeset
690 :param fRunRepresentative: Run Representative based methods (true indicates run).
sagun98
parents:
diff changeset
691 :type: Boolean
sagun98
parents:
diff changeset
692 :param fRunExtreme: Run Extreme based methods (true indicates run).
sagun98
parents:
diff changeset
693 :type: Boolean
sagun98
parents:
diff changeset
694 :param istmBetaMatrix: File that has a precalculated beta matrix
sagun98
parents:
diff changeset
695 :type: File stream or File path string
sagun98
parents:
diff changeset
696 :return Selected Samples: Samples selected by methods.
sagun98
parents:
diff changeset
697 Dictionary {"Selection Method":["SampleID","SampleID","SampleID",...]}
sagun98
parents:
diff changeset
698 """
sagun98
parents:
diff changeset
699
sagun98
parents:
diff changeset
700 #Sample ids/names
sagun98
parents:
diff changeset
701 lsSampleNames = abndData.funcGetSampleNames()
sagun98
parents:
diff changeset
702
sagun98
parents:
diff changeset
703 #Generate alpha metrics and get most diverse
sagun98
parents:
diff changeset
704 if fRunDiversity:
sagun98
parents:
diff changeset
705
sagun98
parents:
diff changeset
706 #Get Alpha metrics matrix
sagun98
parents:
diff changeset
707 internalAlphaMatrix = None
sagun98
parents:
diff changeset
708 #Name of technique
sagun98
parents:
diff changeset
709 strMethod = [strAlphaMetadata] if strAlphaMetadata else lsAlphaMetrics
sagun98
parents:
diff changeset
710
sagun98
parents:
diff changeset
711 #If given an alpha-diversity metadata
sagun98
parents:
diff changeset
712 if strAlphaMetadata:
sagun98
parents:
diff changeset
713 internalAlphaMatrix = [[float(strNum) for strNum in abndData.funcGetMetadata(strAlphaMetadata)]]
sagun98
parents:
diff changeset
714 else:
sagun98
parents:
diff changeset
715 #Expects Observations (Taxa (row) x sample (column))
sagun98
parents:
diff changeset
716 #Returns [[metric1-sample1, metric1-sample2, metric1-sample3],[metric1-sample1, metric1-sample2, metric1-sample3]]
sagun98
parents:
diff changeset
717 internalAlphaMatrix = Metric.funcBuildAlphaMetricsMatrix(npaSampleAbundance = abndData.funcGetAbundanceCopy()
sagun98
parents:
diff changeset
718 if not abndData.funcIsSummed()
sagun98
parents:
diff changeset
719 else abndData.funcGetFeatureAbundanceTable(abndData.funcGetTerminalNodes()).funcGetAbundanceCopy(),
sagun98
parents:
diff changeset
720 lsSampleNames = lsSampleNames, lsDiversityMetricAlpha = lsAlphaMetrics)
sagun98
parents:
diff changeset
721
sagun98
parents:
diff changeset
722 if internalAlphaMatrix:
sagun98
parents:
diff changeset
723 #Invert measurments
sagun98
parents:
diff changeset
724 if fInvertDiversity:
sagun98
parents:
diff changeset
725 lldNewDiversity = []
sagun98
parents:
diff changeset
726 for lsLine in internalAlphaMatrix:
sagun98
parents:
diff changeset
727 lldNewDiversity.append([1/max(dValue,ConstantsMicropita.c_smallNumber) for dValue in lsLine])
sagun98
parents:
diff changeset
728 internalAlphaMatrix = lldNewDiversity
sagun98
parents:
diff changeset
729 #Get top ranked alpha diversity by most diverse
sagun98
parents:
diff changeset
730 #Expects [[sample1,sample2,sample3...],[sample1,sample2,sample3..],...]
sagun98
parents:
diff changeset
731 #Returns [[sampleName1, sampleName2, sampleNameN],[sampleName1, sampleName2, sampleNameN]]
sagun98
parents:
diff changeset
732 mostDiverseAlphaSamplesIndexes = self.funcGetTopRankedSamples(lldMatrix=internalAlphaMatrix, lsSampleNames=lsSampleNames, iTopAmount=iSampleSelectionCount)
sagun98
parents:
diff changeset
733
sagun98
parents:
diff changeset
734 #Add to results
sagun98
parents:
diff changeset
735 for index in xrange(0,len(strMethod)):
sagun98
parents:
diff changeset
736 strSelectionMethod = self.dictConvertAMetricDiversity.get(strMethod[index],ConstantsMicropita.c_strDiversity+"="+strMethod[index])
sagun98
parents:
diff changeset
737 dictSelectedSamples.setdefault(strSelectionMethod,[]).extend(mostDiverseAlphaSamplesIndexes[index])
sagun98
parents:
diff changeset
738
sagun98
parents:
diff changeset
739 logging.info("MicroPITA.funcRunNormalizeSensitiveMethods:: Selected Samples 1b")
sagun98
parents:
diff changeset
740 logging.info(dictSelectedSamples)
sagun98
parents:
diff changeset
741
sagun98
parents:
diff changeset
742 #Generate beta metrics and
sagun98
parents:
diff changeset
743 if fRunRepresentative or fRunExtreme:
sagun98
parents:
diff changeset
744
sagun98
parents:
diff changeset
745 #Abundance matrix transposed
sagun98
parents:
diff changeset
746 npaTransposedAbundance = UtilityMath.funcTransposeDataMatrix(abndData.funcGetAbundanceCopy(), fRemoveAdornments=True)
sagun98
parents:
diff changeset
747
sagun98
parents:
diff changeset
748 #Get center selection using clusters/tiling
sagun98
parents:
diff changeset
749 #This will be for beta metrics in normalized space
sagun98
parents:
diff changeset
750 if fRunRepresentative:
sagun98
parents:
diff changeset
751
sagun98
parents:
diff changeset
752 if istmBetaMatrix:
sagun98
parents:
diff changeset
753 #Get representative dissimilarity samples
sagun98
parents:
diff changeset
754 medoidSamples=self.funcGetCentralSamplesByKMedoids(npaMatrix=npaTransposedAbundance, sMetric=ConstantsMicropita.c_custom, lsSampleNames=lsSampleNames, iNumberSamplesReturned=iSampleSelectionCount, istmBetaMatrix=istmBetaMatrix, istrmTree=istrmTree, istrmEnvr=istrmEnvr)
sagun98
parents:
diff changeset
755
sagun98
parents:
diff changeset
756 if medoidSamples:
sagun98
parents:
diff changeset
757 dictSelectedSamples.setdefault(ConstantsMicropita.c_strRepresentative+"="+ConstantsMicropita.c_custom,[]).extend(medoidSamples)
sagun98
parents:
diff changeset
758 else:
sagun98
parents:
diff changeset
759 logging.info("MicroPITA.funcRunNormalizeSensitiveMethods:: Performing representative selection on normalized data.")
sagun98
parents:
diff changeset
760 for bMetric in lsBetaMetrics:
sagun98
parents:
diff changeset
761
sagun98
parents:
diff changeset
762 #Get representative dissimilarity samples
sagun98
parents:
diff changeset
763 medoidSamples=self.funcGetCentralSamplesByKMedoids(npaMatrix=npaTransposedAbundance, sMetric=bMetric, lsSampleNames=lsSampleNames, iNumberSamplesReturned=iSampleSelectionCount, istmBetaMatrix=istmBetaMatrix, istrmTree=istrmTree, istrmEnvr=istrmEnvr)
sagun98
parents:
diff changeset
764
sagun98
parents:
diff changeset
765 if medoidSamples:
sagun98
parents:
diff changeset
766 dictSelectedSamples.setdefault(self.dictConvertBMetricToMethod.get(bMetric,ConstantsMicropita.c_strRepresentative+"="+bMetric),[]).extend(medoidSamples)
sagun98
parents:
diff changeset
767
sagun98
parents:
diff changeset
768 #Get extreme selection using clusters, tiling
sagun98
parents:
diff changeset
769 if fRunExtreme:
sagun98
parents:
diff changeset
770 logging.info("MicroPITA.funcRunNormalizeSensitiveMethods:: Performing extreme selection on normalized data.")
sagun98
parents:
diff changeset
771 if istmBetaMatrix:
sagun98
parents:
diff changeset
772
sagun98
parents:
diff changeset
773 #Samples for representative dissimilarity
sagun98
parents:
diff changeset
774 #This involves inverting the distance metric,
sagun98
parents:
diff changeset
775 #Taking the dendrogram level of where the number cluster == the number of samples to select
sagun98
parents:
diff changeset
776 #Returning a repersentative sample from each cluster
sagun98
parents:
diff changeset
777 extremeSamples = self.funcSelectExtremeSamplesFromHClust(strBetaMetric=ConstantsMicropita.c_custom, npaAbundanceMatrix=npaTransposedAbundance, lsSampleNames=lsSampleNames, iSelectSampleCount=iSampleSelectionCount, istmBetaMatrix=istmBetaMatrix, istrmTree=istrmTree, istrmEnvr=istrmEnvr)
sagun98
parents:
diff changeset
778
sagun98
parents:
diff changeset
779 #Add selected samples
sagun98
parents:
diff changeset
780 if extremeSamples:
sagun98
parents:
diff changeset
781 dictSelectedSamples.setdefault(ConstantsMicropita.c_strExtreme+"="+ConstantsMicropita.c_custom,[]).extend(extremeSamples)
sagun98
parents:
diff changeset
782
sagun98
parents:
diff changeset
783 else:
sagun98
parents:
diff changeset
784 #Run KMedoids with inverse custom distance metric in normalized space
sagun98
parents:
diff changeset
785 for bMetric in lsInverseBetaMetrics:
sagun98
parents:
diff changeset
786
sagun98
parents:
diff changeset
787 #Samples for representative dissimilarity
sagun98
parents:
diff changeset
788 #This involves inverting the distance metric,
sagun98
parents:
diff changeset
789 #Taking the dendrogram level of where the number cluster == the number of samples to select
sagun98
parents:
diff changeset
790 #Returning a repersentative sample from each cluster
sagun98
parents:
diff changeset
791 extremeSamples = self.funcSelectExtremeSamplesFromHClust(strBetaMetric=bMetric, npaAbundanceMatrix=npaTransposedAbundance, lsSampleNames=lsSampleNames, iSelectSampleCount=iSampleSelectionCount, istmBetaMatrix=istmBetaMatrix, istrmTree=istrmTree, istrmEnvr=istrmEnvr)
sagun98
parents:
diff changeset
792
sagun98
parents:
diff changeset
793 #Add selected samples
sagun98
parents:
diff changeset
794 if extremeSamples:
sagun98
parents:
diff changeset
795 dictSelectedSamples.setdefault(self.dictConvertInvBMetricToMethod.get(bMetric,ConstantsMicropita.c_strExtreme+"="+bMetric),[]).extend(extremeSamples)
sagun98
parents:
diff changeset
796
sagun98
parents:
diff changeset
797 logging.info("MicroPITA.funcRunNormalizeSensitiveMethods:: Selected Samples 2,3b")
sagun98
parents:
diff changeset
798 logging.info(dictSelectedSamples)
sagun98
parents:
diff changeset
799 return dictSelectedSamples
sagun98
parents:
diff changeset
800
sagun98
parents:
diff changeset
801 def funcRun(self, strIDName, strLastMetadataName, istmInput,
sagun98
parents:
diff changeset
802 ostmInputPredictFile, ostmPredictFile, ostmCheckedFile, ostmOutput,
sagun98
parents:
diff changeset
803 cDelimiter, cFeatureNameDelimiter, strFeatureSelection,
sagun98
parents:
diff changeset
804 istmFeatures, iCount, lstrMethods, strLastRowMetadata = None, strLabel = None, strStratify = None,
sagun98
parents:
diff changeset
805 strCustomAlpha = None, strCustomBeta = None, strAlphaMetadata = None, istmBetaMatrix = None, istrmTree = None, istrmEnvr = None,
sagun98
parents:
diff changeset
806 iMinSeqs = ConstantsMicropita.c_liOccurenceFilter[0], iMinSamples = ConstantsMicropita.c_liOccurenceFilter[1], fInvertDiversity = False):
sagun98
parents:
diff changeset
807 """
sagun98
parents:
diff changeset
808 Manages the selection of samples given different metrics.
sagun98
parents:
diff changeset
809
sagun98
parents:
diff changeset
810 :param strIDName: Sample Id metadata row
sagun98
parents:
diff changeset
811 :type: String
sagun98
parents:
diff changeset
812 :param strLastMetadataName: The id of the metadata positioned last in the abundance table.
sagun98
parents:
diff changeset
813 :type: String String metadata id.
sagun98
parents:
diff changeset
814 :param istmInput: File to store input data to supervised methods.
sagun98
parents:
diff changeset
815 :type: FileStream of String file path
sagun98
parents:
diff changeset
816 :param ostmInputPredictFile: File to store distances from supervised methods.
sagun98
parents:
diff changeset
817 :type: FileStream or String file path
sagun98
parents:
diff changeset
818 :param ostmCheckedFile: File to store the AbundanceTable data after it is being checked.
sagun98
parents:
diff changeset
819 :type: FileStream or String file path
sagun98
parents:
diff changeset
820 :param ostmOutPut: File to store sample selection by methods of interest.
sagun98
parents:
diff changeset
821 :type: FileStream or String file path
sagun98
parents:
diff changeset
822 :param cDelimiter: Delimiter of abundance table.
sagun98
parents:
diff changeset
823 :type: Character Char (default TAB).
sagun98
parents:
diff changeset
824 :param cFeatureNameDelimiter: Delimiter of the name of features (for instance if they contain consensus lineages indicating clades).
sagun98
parents:
diff changeset
825 :type: Character (default |).
sagun98
parents:
diff changeset
826 :param stFeatureSelectionMethod: Which method to use to select features in a targeted manner (Using average ranked abundance or average abundance).
sagun98
parents:
diff changeset
827 :type: String (specific values indicated in ConstantsMicropita.lsTargetedFeatureMethodValues).
sagun98
parents:
diff changeset
828 :param istmFeatures: File which holds the features of interest if using targeted feature methodology.
sagun98
parents:
diff changeset
829 :type: FileStream or String file path
sagun98
parents:
diff changeset
830 :param iCount: Number of samples to select in each methods, supervised methods select this amount per label if possible.
sagun98
parents:
diff changeset
831 :type: Integer integer.
sagun98
parents:
diff changeset
832 :param lstrMethods: List of strings indicating selection techniques.
sagun98
parents:
diff changeset
833 :type: List of string method names
sagun98
parents:
diff changeset
834 :param strLabel: The metadata used for supervised labels.
sagun98
parents:
diff changeset
835 :type: String
sagun98
parents:
diff changeset
836 :param strStratify: The metadata used to stratify unsupervised data.
sagun98
parents:
diff changeset
837 :type: String
sagun98
parents:
diff changeset
838 :param strCustomAlpha: Custom alpha diversity metric
sagun98
parents:
diff changeset
839 :type: String
sagun98
parents:
diff changeset
840 :param strCustomBeta: Custom beta diversity metric
sagun98
parents:
diff changeset
841 :type: String
sagun98
parents:
diff changeset
842 :param strAlphaMetadata: Metadata id which is a diveristy metric to use in highest diversity sampling
sagun98
parents:
diff changeset
843 :type: String
sagun98
parents:
diff changeset
844 :param istmBetaMatrix: File containing precalculated beta-diversity matrix for representative sampling
sagun98
parents:
diff changeset
845 :type: FileStream or String file path
sagun98
parents:
diff changeset
846 :param istrmTree: File containing tree for phylogentic beta-diversity analysis
sagun98
parents:
diff changeset
847 :type: FileStream or String file path
sagun98
parents:
diff changeset
848 :param istrmEnvr: File containing environment for phylogentic beta-diversity analysis
sagun98
parents:
diff changeset
849 :type: FileStream or String file path
sagun98
parents:
diff changeset
850 :param iMinSeqs: Minimum sequence in the occurence filter which filters all features not with a minimum number of sequences in each of a minimum number of samples.
sagun98
parents:
diff changeset
851 :type: Integer
sagun98
parents:
diff changeset
852 :param iMinSamples: Minimum sample count for the occurence filter.
sagun98
parents:
diff changeset
853 :type: Integer
sagun98
parents:
diff changeset
854 :param fInvertDiversity: When true will invert diversity measurements before using.
sagun98
parents:
diff changeset
855 :type: boolean
sagun98
parents:
diff changeset
856 :return Selected Samples: Samples selected by methods.
sagun98
parents:
diff changeset
857 Dictionary {"Selection Method":["SampleID","SampleID","SampleID",...]}
sagun98
parents:
diff changeset
858 """
sagun98
parents:
diff changeset
859
sagun98
parents:
diff changeset
860 #Holds the top ranked samples from different metrics
sagun98
parents:
diff changeset
861 #dict[metric name] = [samplename,samplename...]
sagun98
parents:
diff changeset
862 selectedSamples = dict()
sagun98
parents:
diff changeset
863
sagun98
parents:
diff changeset
864 #If a target feature file is given make sure that targeted feature is in the selection methods, if not add
sagun98
parents:
diff changeset
865 if ConstantsMicropita.c_strFeature in lstrMethods:
sagun98
parents:
diff changeset
866 if not istmFeatures:
sagun98
parents:
diff changeset
867 logging.error("MicroPITA.funcRun:: Did not receive both the Targeted feature file and the feature selection method. MicroPITA did not run.")
sagun98
parents:
diff changeset
868 return False
sagun98
parents:
diff changeset
869
sagun98
parents:
diff changeset
870 #Diversity metrics to run
sagun98
parents:
diff changeset
871 #Use custom metrics if specified
sagun98
parents:
diff changeset
872 #Custom beta metrics set to normalized only, custom alpha metrics set to count only
sagun98
parents:
diff changeset
873 diversityMetricsAlpha = [] if strCustomAlpha or strAlphaMetadata else [MicroPITA.c_strInverseSimpsonDiversity]
sagun98
parents:
diff changeset
874 diversityMetricsBeta = [] if istmBetaMatrix else [strCustomBeta] if strCustomBeta else [MicroPITA.c_strBrayCurtisDissimilarity]
sagun98
parents:
diff changeset
875 # inverseDiversityMetricsBeta = [MicroPITA.c_strInvBrayCurtisDissimilarity]
sagun98
parents:
diff changeset
876 diversityMetricsAlphaNoNormalize = [strAlphaMetadata] if strAlphaMetadata else [strCustomAlpha] if strCustomAlpha else []
sagun98
parents:
diff changeset
877 diversityMetricsBetaNoNormalize = []
sagun98
parents:
diff changeset
878 # inverseDiversityMetricsBetaNoNormalize = []
sagun98
parents:
diff changeset
879
sagun98
parents:
diff changeset
880 #Targeted taxa
sagun98
parents:
diff changeset
881 userDefinedTaxa = []
sagun98
parents:
diff changeset
882
sagun98
parents:
diff changeset
883 #Perform different flows flags
sagun98
parents:
diff changeset
884 c_RUN_MAX_DIVERSITY_1 = ConstantsMicropita.c_strDiversity in lstrMethods
sagun98
parents:
diff changeset
885 c_RUN_REPRESENTIVE_DISSIMILARITY_2 = ConstantsMicropita.c_strRepresentative in lstrMethods
sagun98
parents:
diff changeset
886 c_RUN_MAX_DISSIMILARITY_3 = ConstantsMicropita.c_strExtreme in lstrMethods
sagun98
parents:
diff changeset
887 c_RUN_RANK_AVERAGE_USER_4 = False
sagun98
parents:
diff changeset
888 if ConstantsMicropita.c_strFeature in lstrMethods:
sagun98
parents:
diff changeset
889 c_RUN_RANK_AVERAGE_USER_4 = True
sagun98
parents:
diff changeset
890 if not istmFeatures:
sagun98
parents:
diff changeset
891 logging.error("MicroPITA.funcRun:: No taxa file was given for taxa selection.")
sagun98
parents:
diff changeset
892 return False
sagun98
parents:
diff changeset
893 #Read in taxa list, break down to lines and filter out empty strings
sagun98
parents:
diff changeset
894 userDefinedTaxa = filter(None,(s.strip( ) for s in istmFeatures.readlines()))
sagun98
parents:
diff changeset
895 c_RUN_RANDOM_5 = ConstantsMicropita.c_strRandom in lstrMethods
sagun98
parents:
diff changeset
896 c_RUN_DISTINCT = ConstantsMicropita.c_strDistinct in lstrMethods
sagun98
parents:
diff changeset
897 c_RUN_DISCRIMINANT = ConstantsMicropita.c_strDiscriminant in lstrMethods
sagun98
parents:
diff changeset
898
sagun98
parents:
diff changeset
899 #Read in abundance data
sagun98
parents:
diff changeset
900 #Abundance is a structured array. Samples (column) by Taxa (rows) with the taxa id row included as the column index=0
sagun98
parents:
diff changeset
901 #Abundance table object to read in and manage data
sagun98
parents:
diff changeset
902 totalAbundanceTable = AbundanceTable.funcMakeFromFile(xInputFile=istmInput, lOccurenceFilter = [iMinSeqs, iMinSamples],
sagun98
parents:
diff changeset
903 cDelimiter=cDelimiter, sMetadataID=strIDName, sLastMetadataRow=strLastRowMetadata,
sagun98
parents:
diff changeset
904 sLastMetadata=strLastMetadataName, cFeatureNameDelimiter=cFeatureNameDelimiter, xOutputFile=ostmCheckedFile)
sagun98
parents:
diff changeset
905 if not totalAbundanceTable:
sagun98
parents:
diff changeset
906 logging.error("MicroPITA.funcRun:: Could not read in the abundance table. Analysis was not performed."+
sagun98
parents:
diff changeset
907 " This often occurs when the Last Metadata is not specified correctly."+
sagun98
parents:
diff changeset
908 " Please check to make sure the Last Metadata selection is the row of the last metadata,"+
sagun98
parents:
diff changeset
909 " all values after this selection should be microbial measurements and should be numeric.")
sagun98
parents:
diff changeset
910 return False
sagun98
parents:
diff changeset
911
sagun98
parents:
diff changeset
912 lsOriginalLabels = SVM.funcMakeLabels(totalAbundanceTable.funcGetMetadata(strLabel)) if strLabel else strLabel
sagun98
parents:
diff changeset
913
sagun98
parents:
diff changeset
914 dictTotalMetadata = totalAbundanceTable.funcGetMetadataCopy()
sagun98
parents:
diff changeset
915 logging.debug("MicroPITA.funcRun:: Received metadata=" + str(dictTotalMetadata))
sagun98
parents:
diff changeset
916 #If there is only 1 unique value for the labels, do not run the Supervised methods
sagun98
parents:
diff changeset
917 if strLabel and ( len(set(dictTotalMetadata.get(strLabel,[]))) < 2 ):
sagun98
parents:
diff changeset
918 logging.error("The label " + strLabel + " did not have 2 or more values. Labels found=" + str(dictTotalMetadata.get(strLabel,[])))
sagun98
parents:
diff changeset
919 return False
sagun98
parents:
diff changeset
920
sagun98
parents:
diff changeset
921 #Run unsupervised methods###
sagun98
parents:
diff changeset
922 #Stratify the data if need be and drop the old data
sagun98
parents:
diff changeset
923 lStratifiedAbundanceTables = totalAbundanceTable.funcStratifyByMetadata(strStratify) if strStratify else [totalAbundanceTable]
sagun98
parents:
diff changeset
924
sagun98
parents:
diff changeset
925 #For each stratified abundance block or for the unstratfified abundance
sagun98
parents:
diff changeset
926 #Run the unsupervised blocks
sagun98
parents:
diff changeset
927 fAppendSupFiles = False
sagun98
parents:
diff changeset
928 for stratAbundanceTable in lStratifiedAbundanceTables:
sagun98
parents:
diff changeset
929 logging.info("MicroPITA.funcRun:: Running abundance block:"+stratAbundanceTable.funcGetName())
sagun98
parents:
diff changeset
930
sagun98
parents:
diff changeset
931 ###NOT SUMMED, NOT NORMALIZED
sagun98
parents:
diff changeset
932 #Only perform if the data is not yet normalized
sagun98
parents:
diff changeset
933 if not stratAbundanceTable.funcIsNormalized( ):
sagun98
parents:
diff changeset
934 #Need to first work with unnormalized data
sagun98
parents:
diff changeset
935 if c_RUN_MAX_DIVERSITY_1 or c_RUN_REPRESENTIVE_DISSIMILARITY_2 or c_RUN_MAX_DISSIMILARITY_3:
sagun98
parents:
diff changeset
936
sagun98
parents:
diff changeset
937 self._funcRunNormalizeSensitiveMethods(abndData=stratAbundanceTable, iSampleSelectionCount=iCount,
sagun98
parents:
diff changeset
938 dictSelectedSamples=selectedSamples, lsAlphaMetrics=diversityMetricsAlphaNoNormalize,
sagun98
parents:
diff changeset
939 lsBetaMetrics=diversityMetricsBetaNoNormalize,
sagun98
parents:
diff changeset
940 lsInverseBetaMetrics=diversityMetricsBetaNoNormalize,
sagun98
parents:
diff changeset
941 fRunDiversity=c_RUN_MAX_DIVERSITY_1,fRunRepresentative=c_RUN_REPRESENTIVE_DISSIMILARITY_2,
sagun98
parents:
diff changeset
942 fRunExtreme=c_RUN_MAX_DISSIMILARITY_3, strAlphaMetadata=strAlphaMetadata,
sagun98
parents:
diff changeset
943 istrmTree=istrmTree, istrmEnvr=istrmEnvr, fInvertDiversity=fInvertDiversity)
sagun98
parents:
diff changeset
944
sagun98
parents:
diff changeset
945
sagun98
parents:
diff changeset
946 #Generate selection by the rank average of user defined taxa
sagun98
parents:
diff changeset
947 #Expects (Taxa (row) by Samples (column))
sagun98
parents:
diff changeset
948 #Expects a column 0 of taxa id that is skipped
sagun98
parents:
diff changeset
949 #Returns [(sample name,average,rank)]
sagun98
parents:
diff changeset
950 #SUMMED AND NORMALIZED
sagun98
parents:
diff changeset
951 stratAbundanceTable.funcSumClades()
sagun98
parents:
diff changeset
952 #Normalize data at this point
sagun98
parents:
diff changeset
953 stratAbundanceTable.funcNormalize()
sagun98
parents:
diff changeset
954 if c_RUN_RANK_AVERAGE_USER_4:
sagun98
parents:
diff changeset
955 selectedSamples[ConstantsMicropita.c_strFeature] = self.funcSelectTargetedTaxaSamples(abndMatrix=stratAbundanceTable,
sagun98
parents:
diff changeset
956 lsTargetedTaxa=userDefinedTaxa, iSampleSelectionCount=iCount, sMethod=strFeatureSelection)
sagun98
parents:
diff changeset
957 logging.info("MicroPITA.funcRun:: Selected Samples Rank")
sagun98
parents:
diff changeset
958 logging.info(selectedSamples)
sagun98
parents:
diff changeset
959
sagun98
parents:
diff changeset
960 ###SUMMED AND NORMALIZED analysis block
sagun98
parents:
diff changeset
961 #Diversity based metric will move reduce to terminal taxa as needed
sagun98
parents:
diff changeset
962 if c_RUN_MAX_DIVERSITY_1 or c_RUN_REPRESENTIVE_DISSIMILARITY_2 or c_RUN_MAX_DISSIMILARITY_3:
sagun98
parents:
diff changeset
963
sagun98
parents:
diff changeset
964 self._funcRunNormalizeSensitiveMethods(abndData=stratAbundanceTable, iSampleSelectionCount=iCount,
sagun98
parents:
diff changeset
965 dictSelectedSamples=selectedSamples, lsAlphaMetrics=diversityMetricsAlpha,
sagun98
parents:
diff changeset
966 lsBetaMetrics=diversityMetricsBeta,
sagun98
parents:
diff changeset
967 lsInverseBetaMetrics=diversityMetricsBeta,
sagun98
parents:
diff changeset
968 fRunDiversity=c_RUN_MAX_DIVERSITY_1,fRunRepresentative=c_RUN_REPRESENTIVE_DISSIMILARITY_2,
sagun98
parents:
diff changeset
969 fRunExtreme=c_RUN_MAX_DISSIMILARITY_3,
sagun98
parents:
diff changeset
970 istmBetaMatrix=istmBetaMatrix, istrmTree=istrmTree, istrmEnvr=istrmEnvr, fInvertDiversity=fInvertDiversity)
sagun98
parents:
diff changeset
971
sagun98
parents:
diff changeset
972 #5::Select randomly
sagun98
parents:
diff changeset
973 #Expects sampleNames = List of sample names [name, name, name...]
sagun98
parents:
diff changeset
974 if(c_RUN_RANDOM_5):
sagun98
parents:
diff changeset
975 #Select randomly from sample names
sagun98
parents:
diff changeset
976 selectedSamples[ConstantsMicropita.c_strRandom] = self.funcGetRandomSamples(lsSamples=stratAbundanceTable.funcGetSampleNames(), iNumberOfSamplesToReturn=iCount)
sagun98
parents:
diff changeset
977 logging.info("MicroPITA.funcRun:: Selected Samples Random")
sagun98
parents:
diff changeset
978 logging.info(selectedSamples)
sagun98
parents:
diff changeset
979
sagun98
parents:
diff changeset
980 #Perform supervised selection
sagun98
parents:
diff changeset
981 if c_RUN_DISTINCT or c_RUN_DISCRIMINANT:
sagun98
parents:
diff changeset
982 if strLabel:
sagun98
parents:
diff changeset
983 dictSelectionRet = self.funcRunSupervisedDistancesFromCentroids(abundanceTable=stratAbundanceTable,
sagun98
parents:
diff changeset
984 fRunDistinct=c_RUN_DISTINCT, fRunDiscriminant=c_RUN_DISCRIMINANT,
sagun98
parents:
diff changeset
985 xOutputSupFile=ostmInputPredictFile,xPredictSupFile=ostmPredictFile,
sagun98
parents:
diff changeset
986 strSupervisedMetadata=strLabel, iSampleSupSelectionCount=iCount,
sagun98
parents:
diff changeset
987 lsOriginalSampleNames = totalAbundanceTable.funcGetSampleNames(),
sagun98
parents:
diff changeset
988 lsOriginalLabels = lsOriginalLabels,
sagun98
parents:
diff changeset
989 fAppendFiles=fAppendSupFiles)
sagun98
parents:
diff changeset
990
sagun98
parents:
diff changeset
991 [selectedSamples.setdefault(sKey,[]).extend(lValue) for sKey,lValue in dictSelectionRet.items()]
sagun98
parents:
diff changeset
992
sagun98
parents:
diff changeset
993 if not fAppendSupFiles:
sagun98
parents:
diff changeset
994 fAppendSupFiles = True
sagun98
parents:
diff changeset
995 logging.info("MicroPITA.funcRun:: Selected Samples Unsupervised")
sagun98
parents:
diff changeset
996 logging.info(selectedSamples)
sagun98
parents:
diff changeset
997 return selectedSamples
sagun98
parents:
diff changeset
998
sagun98
parents:
diff changeset
999 #Testing: Happy path tested
sagun98
parents:
diff changeset
1000 @staticmethod
sagun98
parents:
diff changeset
1001 def funcWriteSelectionToFile(dictSelection,xOutputFilePath):
sagun98
parents:
diff changeset
1002 """
sagun98
parents:
diff changeset
1003 Writes the selection of samples by method to an output file.
sagun98
parents:
diff changeset
1004
sagun98
parents:
diff changeset
1005 :param dictSelection: The dictionary of selections by method to be written to a file.
sagun98
parents:
diff changeset
1006 :type: Dictionary The dictionary of selections by method {"method":["sample selected","sample selected"...]}
sagun98
parents:
diff changeset
1007 :param xOutputFilePath: FileStream or String path to file inwhich the dictionary is written.
sagun98
parents:
diff changeset
1008 :type: String FileStream or String path to file
sagun98
parents:
diff changeset
1009 """
sagun98
parents:
diff changeset
1010
sagun98
parents:
diff changeset
1011 if not dictSelection:
sagun98
parents:
diff changeset
1012 return
sagun98
parents:
diff changeset
1013
sagun98
parents:
diff changeset
1014 #Open file
sagun98
parents:
diff changeset
1015 f = csv.writer(open(xOutputFilePath,"w") if isinstance(xOutputFilePath, str) else xOutputFilePath, delimiter=ConstantsMicropita.c_outputFileDelim )
sagun98
parents:
diff changeset
1016
sagun98
parents:
diff changeset
1017 #Create output content from dictionary
sagun98
parents:
diff changeset
1018 for sKey in dictSelection:
sagun98
parents:
diff changeset
1019 f.writerow([sKey]+dictSelection[sKey])
sagun98
parents:
diff changeset
1020 logging.debug("MicroPITA.funcRun:: Selected samples output to file:"+str(dictSelection[sKey]))
sagun98
parents:
diff changeset
1021
sagun98
parents:
diff changeset
1022 #Testing: Happy Path tested
sagun98
parents:
diff changeset
1023 @staticmethod
sagun98
parents:
diff changeset
1024 def funcReadSelectionFileToDictionary(xInputFile):
sagun98
parents:
diff changeset
1025 """
sagun98
parents:
diff changeset
1026 Reads in an output selection file from micropita and formats it into a dictionary.
sagun98
parents:
diff changeset
1027
sagun98
parents:
diff changeset
1028 :param xInputFile: String path to file or file stream to read and translate into a dictionary.
sagun98
parents:
diff changeset
1029 {"method":["sample selected","sample selected"...]}
sagun98
parents:
diff changeset
1030 :type: FileStream or String Path to file
sagun98
parents:
diff changeset
1031 :return Dictionary: Samples selected by methods.
sagun98
parents:
diff changeset
1032 Dictionary {"Selection Method":["SampleID","SampleID","SampleID",...]}
sagun98
parents:
diff changeset
1033 """
sagun98
parents:
diff changeset
1034
sagun98
parents:
diff changeset
1035 #Open file
sagun98
parents:
diff changeset
1036 istmReader = csv.reader(open(xInputFile,'r') if isinstance(xInputFile, str) else xInputFile, delimiter = ConstantsMicropita.c_outputFileDelim)
sagun98
parents:
diff changeset
1037
sagun98
parents:
diff changeset
1038 #Dictionary to hold selection data
sagun98
parents:
diff changeset
1039 return dict([(lsLine[0], lsLine[1:]) for lsLine in istmReader])
sagun98
parents:
diff changeset
1040
sagun98
parents:
diff changeset
1041 #Set up arguments reader
sagun98
parents:
diff changeset
1042 argp = argparse.ArgumentParser( prog = "MicroPITA.py",
sagun98
parents:
diff changeset
1043 description = """Selects samples from abundance tables based on various selection schemes.""" )
sagun98
parents:
diff changeset
1044
sagun98
parents:
diff changeset
1045 args = argp.add_argument_group( "Common", "Commonly modified options" )
sagun98
parents:
diff changeset
1046 args.add_argument(ConstantsMicropita.c_strCountArgument,"--num", dest="iCount", metavar = "samples", default = 10, type = int, help = ConstantsMicropita.c_strCountHelp)
sagun98
parents:
diff changeset
1047 args.add_argument("-m","--method", dest = "lstrMethods", metavar = "method", default = [], help = ConstantsMicropita.c_strSelectionTechniquesHelp,
sagun98
parents:
diff changeset
1048 choices = ConstantsMicropita.c_lsAllMethods, action = "append")
sagun98
parents:
diff changeset
1049
sagun98
parents:
diff changeset
1050 args = argp.add_argument_group( "Custom", "Selecting and inputing custom metrics" )
sagun98
parents:
diff changeset
1051 args.add_argument("-a","--alpha", dest = "strAlphaDiversity", metavar = "AlphaDiversity", default = None, help = ConstantsMicropita.c_strCustomAlphaDiversityHelp, choices = Metric.setAlphaDiversities)
sagun98
parents:
diff changeset
1052 args.add_argument("-b","--beta", dest = "strBetaDiversity", metavar = "BetaDiversity", default = None, help = ConstantsMicropita.c_strCustomBetaDiversityHelp, choices = list(Metric.setBetaDiversities)+[Metric.c_strUnifracUnweighted,Metric.c_strUnifracWeighted])
sagun98
parents:
diff changeset
1053 args.add_argument("-q","--alphameta", dest = "strAlphaMetadata", metavar = "AlphaDiversityMetadata", default = None, help = ConstantsMicropita.c_strCustomAlphaDiversityMetadataHelp)
sagun98
parents:
diff changeset
1054 args.add_argument("-x","--betamatrix", dest = "istmBetaMatrix", metavar = "BetaDiversityMatrix", default = None, help = ConstantsMicropita.c_strCustomBetaDiversityMatrixHelp)
sagun98
parents:
diff changeset
1055 args.add_argument("-o","--tree", dest = "istrmTree", metavar = "PhylogeneticTree", default = None, help = ConstantsMicropita.c_strCustomPhylogeneticTreeHelp)
sagun98
parents:
diff changeset
1056 args.add_argument("-i","--envr", dest = "istrmEnvr", metavar = "EnvironmentFile", default = None, help = ConstantsMicropita.c_strCustomEnvironmentFileHelp)
sagun98
parents:
diff changeset
1057 args.add_argument("-f","--invertDiversity", dest = "fInvertDiversity", action="store_true", default = False, help = ConstantsMicropita.c_strInvertDiversityHelp)
sagun98
parents:
diff changeset
1058
sagun98
parents:
diff changeset
1059 args = argp.add_argument_group( "Miscellaneous", "Row/column identifiers and feature targeting options" )
sagun98
parents:
diff changeset
1060 args.add_argument("-d",ConstantsMicropita.c_strIDNameArgument, dest="strIDName", metavar="sample_id", help= ConstantsMicropita.c_strIDNameHelp)
sagun98
parents:
diff changeset
1061 args.add_argument("-l",ConstantsMicropita.c_strLastMetadataNameArgument, dest="strLastMetadataName", metavar = "metadata_id", default = None,
sagun98
parents:
diff changeset
1062 help= ConstantsMicropita.c_strLastMetadataNameHelp)
sagun98
parents:
diff changeset
1063 args.add_argument("-r",ConstantsMicropita.c_strTargetedFeatureMethodArgument, dest="strFeatureSelection", metavar="targeting_method", default=ConstantsMicropita.lsTargetedFeatureMethodValues[0],
sagun98
parents:
diff changeset
1064 choices=ConstantsMicropita.lsTargetedFeatureMethodValues, help= ConstantsMicropita.c_strTargetedFeatureMethodHelp)
sagun98
parents:
diff changeset
1065 args.add_argument("-t",ConstantsMicropita.c_strTargetedSelectionFileArgument, dest="istmFeatures", metavar="feature_file", type=argparse.FileType("rU"), help=ConstantsMicropita.c_strTargetedSelectionFileHelp)
sagun98
parents:
diff changeset
1066 args.add_argument("-w",ConstantsMicropita.c_strFeatureMetadataArgument, dest="strLastFeatureMetadata", metavar="Last_Feature_Metadata", default=None, help=ConstantsMicropita.c_strFeatureMetadataHelp)
sagun98
parents:
diff changeset
1067
sagun98
parents:
diff changeset
1068 args = argp.add_argument_group( "Data labeling", "Metadata IDs for strata and supervised label values" )
sagun98
parents:
diff changeset
1069 args.add_argument("-e",ConstantsMicropita.c_strSupervisedLabelArgument, dest="strLabel", metavar= "supervised_id", help=ConstantsMicropita.c_strSupervisedLabelHelp)
sagun98
parents:
diff changeset
1070 args.add_argument("-s",ConstantsMicropita.c_strUnsupervisedStratifyMetadataArgument, dest="strUnsupervisedStratify", metavar="stratify_id",
sagun98
parents:
diff changeset
1071 help= ConstantsMicropita.c_strUnsupervisedStratifyMetadataHelp)
sagun98
parents:
diff changeset
1072
sagun98
parents:
diff changeset
1073 args = argp.add_argument_group( "File formatting", "Rarely modified file formatting options" )
sagun98
parents:
diff changeset
1074 args.add_argument("-j",ConstantsMicropita.c_strFileDelimiterArgument, dest="cFileDelimiter", metavar="column_delimiter", default="\t", help=ConstantsMicropita.c_strFileDelimiterHelp)
sagun98
parents:
diff changeset
1075 args.add_argument("-k",ConstantsMicropita.c_strFeatureNameDelimiterArgument, dest="cFeatureNameDelimiter", metavar="taxonomy_delimiter", default="|", help=ConstantsMicropita.c_strFeatureNameDelimiterHelp)
sagun98
parents:
diff changeset
1076
sagun98
parents:
diff changeset
1077 args = argp.add_argument_group( "Debugging", "Debugging options - modify at your own risk!" )
sagun98
parents:
diff changeset
1078 args.add_argument("-v",ConstantsMicropita.c_strLoggingArgument, dest="strLogLevel", metavar = "log_level", default="WARNING",
sagun98
parents:
diff changeset
1079 choices=ConstantsMicropita.c_lsLoggingChoices, help= ConstantsMicropita.c_strLoggingHelp)
sagun98
parents:
diff changeset
1080 args.add_argument("-c",ConstantsMicropita.c_strCheckedAbundanceFileArgument, dest="ostmCheckedFile", metavar = "output_qc", type = argparse.FileType("w"), help = ConstantsMicropita.c_strCheckedAbundanceFileHelp)
sagun98
parents:
diff changeset
1081 args.add_argument("-g",ConstantsMicropita.c_strLoggingFileArgument, dest="ostmLoggingFile", metavar = "output_log", type = argparse.FileType("w"), help = ConstantsMicropita.c_strLoggingFileHelp)
sagun98
parents:
diff changeset
1082 args.add_argument("-u",ConstantsMicropita.c_strSupervisedInputFile, dest="ostmInputPredictFile", metavar = "output_scaled", type = argparse.FileType("w"), help = ConstantsMicropita.c_strSupervisedInputFileHelp)
sagun98
parents:
diff changeset
1083 args.add_argument("-p",ConstantsMicropita.c_strSupervisedPredictedFile, dest="ostmPredictFile", metavar = "output_labels", type = argparse.FileType("w"), help = ConstantsMicropita.c_strSupervisedPredictedFileHelp)
sagun98
parents:
diff changeset
1084
sagun98
parents:
diff changeset
1085 argp.add_argument("istmInput", metavar = "input.pcl/biome", type = argparse.FileType("rU"), help = ConstantsMicropita.c_strAbundanceFileHelp,
sagun98
parents:
diff changeset
1086 default = sys.stdin)
sagun98
parents:
diff changeset
1087 argp.add_argument("ostmOutput", metavar = "output.txt", type = argparse.FileType("w"), help = ConstantsMicropita.c_strGenericOutputDataFileHelp,
sagun98
parents:
diff changeset
1088 default = sys.stdout)
sagun98
parents:
diff changeset
1089
sagun98
parents:
diff changeset
1090 __doc__ = "::\n\n\t" + argp.format_help( ).replace( "\n", "\n\t" ) + __doc__
sagun98
parents:
diff changeset
1091
sagun98
parents:
diff changeset
1092 def _main( ):
sagun98
parents:
diff changeset
1093 args = argp.parse_args( )
sagun98
parents:
diff changeset
1094
sagun98
parents:
diff changeset
1095 #Set up logger
sagun98
parents:
diff changeset
1096 iLogLevel = getattr(logging, args.strLogLevel.upper(), None)
sagun98
parents:
diff changeset
1097 logging.basicConfig(stream = args.ostmLoggingFile if args.ostmLoggingFile else sys.stderr, filemode = 'w', level=iLogLevel)
sagun98
parents:
diff changeset
1098
sagun98
parents:
diff changeset
1099 #Run micropita
sagun98
parents:
diff changeset
1100 logging.info("MicroPITA:: Start microPITA")
sagun98
parents:
diff changeset
1101 microPITA = MicroPITA()
sagun98
parents:
diff changeset
1102
sagun98
parents:
diff changeset
1103 #Argparse will append to the default but will not remove the default so I do this here
sagun98
parents:
diff changeset
1104 if not len(args.lstrMethods):
sagun98
parents:
diff changeset
1105 args.lstrMethods = [ConstantsMicropita.c_strRepresentative]
sagun98
parents:
diff changeset
1106
sagun98
parents:
diff changeset
1107 dictSelectedSamples = microPITA.funcRun(
sagun98
parents:
diff changeset
1108 strIDName = args.strIDName,
sagun98
parents:
diff changeset
1109 strLastMetadataName = args.strLastMetadataName,
sagun98
parents:
diff changeset
1110 istmInput = args.istmInput,
sagun98
parents:
diff changeset
1111 ostmInputPredictFile = args.ostmInputPredictFile,
sagun98
parents:
diff changeset
1112 ostmPredictFile = args.ostmPredictFile,
sagun98
parents:
diff changeset
1113 ostmCheckedFile = args.ostmCheckedFile,
sagun98
parents:
diff changeset
1114 ostmOutput = args.ostmOutput,
sagun98
parents:
diff changeset
1115 cDelimiter = args.cFileDelimiter,
sagun98
parents:
diff changeset
1116 cFeatureNameDelimiter = args.cFeatureNameDelimiter,
sagun98
parents:
diff changeset
1117 istmFeatures = args.istmFeatures,
sagun98
parents:
diff changeset
1118 strFeatureSelection = args.strFeatureSelection,
sagun98
parents:
diff changeset
1119 iCount = args.iCount,
sagun98
parents:
diff changeset
1120 strLastRowMetadata = args.strLastFeatureMetadata,
sagun98
parents:
diff changeset
1121 strLabel = args.strLabel,
sagun98
parents:
diff changeset
1122 strStratify = args.strUnsupervisedStratify,
sagun98
parents:
diff changeset
1123 strCustomAlpha = args.strAlphaDiversity,
sagun98
parents:
diff changeset
1124 strCustomBeta = args.strBetaDiversity,
sagun98
parents:
diff changeset
1125 strAlphaMetadata = args.strAlphaMetadata,
sagun98
parents:
diff changeset
1126 istmBetaMatrix = args.istmBetaMatrix,
sagun98
parents:
diff changeset
1127 istrmTree = args.istrmTree,
sagun98
parents:
diff changeset
1128 istrmEnvr = args.istrmEnvr,
sagun98
parents:
diff changeset
1129 lstrMethods = args.lstrMethods,
sagun98
parents:
diff changeset
1130 fInvertDiversity = args.fInvertDiversity
sagun98
parents:
diff changeset
1131 )
sagun98
parents:
diff changeset
1132
sagun98
parents:
diff changeset
1133 if not dictSelectedSamples:
sagun98
parents:
diff changeset
1134 logging.error("MicroPITA:: Error, did not get a result from analysis.")
sagun98
parents:
diff changeset
1135 return -1
sagun98
parents:
diff changeset
1136 logging.info("End microPITA")
sagun98
parents:
diff changeset
1137
sagun98
parents:
diff changeset
1138 #Log output for debugging
sagun98
parents:
diff changeset
1139 logging.debug("MicroPITA:: Returned the following samples:"+str(dictSelectedSamples))
sagun98
parents:
diff changeset
1140
sagun98
parents:
diff changeset
1141 #Write selection to file
sagun98
parents:
diff changeset
1142 microPITA.funcWriteSelectionToFile(dictSelection=dictSelectedSamples, xOutputFilePath=args.ostmOutput)
sagun98
parents:
diff changeset
1143
sagun98
parents:
diff changeset
1144 if __name__ == "__main__":
sagun98
parents:
diff changeset
1145 _main( )