Guest User

Untitled

a guest
Jan 17th, 2018
71
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.29 KB | None | 0 0
  1. model_names = ['is1', 'is5', 'is10', 'im1', 'im5', 'im10']
  2.  
  3. models = [0.1, 0.2, 0.1, 0.3, 0.2, 0.3]
  4.  
  5. dominant_model(models)
  6.  
  7. ['im1', 'im10']
  8.  
  9. def dominant_model(models):
  10. m = max(models)
  11. dom_models = [i for i,j in enumerate(models) if j==m]
  12. for i in range(len(dom_models)):
  13. if dom_models[i]==0:
  14. dom_models[i]=model_names[0]
  15. elif dom_models[i]==1:
  16. dom_models[i]=model_names[1]
  17. elif dom_models[i]==2:
  18. dom_models[i]=model_names[2]
  19. elif dom_models[i]==3:
  20. dom_models[i]=model_names[3]
  21. elif dom_models[i]==4:
  22. dom_models[i]=model_names[4]
  23. elif dom_models[i]==5:
  24. dom_models[i]=model_names[5]
  25. return dom_models
  26.  
  27. def dominant(names, values):
  28. m = max(values)
  29. return [name for name, value in zip(names, values) if value == m]
  30.  
  31. >>> model_names = ['is1', 'is5', 'is10', 'im1', 'im5', 'im10']
  32. >>> model_values = [0.1, 0.2, 0.1, 0.3, 0.2, 0.3]
  33. >>> dominant(model_names, model_values)
  34. ['im1', 'im10']
  35.  
  36. def dominant_model(models):
  37. m = max(models)
  38. dom_models = [i for i,j in enumerate(models) if j==m]
  39. for i in range(len(dom_models)):
  40. for j in range(6):
  41. if dom_models[i]==j:
  42. dom_models[i]=model_names[j]
  43. return dom_models
Add Comment
Please, Sign In to add comment