def findFeature(cat, feat):
if cat == "time":
return [f for f in tds.posFeatures().keys() if f in feat]
if cat == "freq":
return [f for f in fds.posFeatures().keys() if f in feat]
if cat == "vel":
return [f for f in vel.posFeatures().keys() if f in feat]
if cat == "peak":
return [f for f in ps.posFeatures().keys() if f in feat]
def findCategory(el, feat):
if "A" in el: # Time or Freq
if feat in tds.posFeatures().keys():
return "time"
else:
return "freq"
elif "V" in el: # Velocity
return "vel"
return "peak"
def getAllFeatures():
part = {
'time': {
'cols': tds.posCols(),
'features': tds.posFeatures().keys()
},
'freq': {
'cols': fds.posCols(),
'features': fds.posFeatures().keys()
},
'vel': {
'cols': vs.posCols(),
'features': vs.posFeatures().keys()
},
'peak': {
'cols': ps.posPeaks(),
'features': ps.posFeatures().keys()
}
}
return {
'ankle': part,
'hip': part
}
