Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- model_names = ['is1', 'is5', 'is10', 'im1', 'im5', 'im10']
- models = [0.1, 0.2, 0.1, 0.3, 0.2, 0.3]
- dominant_model(models)
- ['im1', 'im10']
- def dominant_model(models):
- m = max(models)
- dom_models = [i for i,j in enumerate(models) if j==m]
- for i in range(len(dom_models)):
- if dom_models[i]==0:
- dom_models[i]=model_names[0]
- elif dom_models[i]==1:
- dom_models[i]=model_names[1]
- elif dom_models[i]==2:
- dom_models[i]=model_names[2]
- elif dom_models[i]==3:
- dom_models[i]=model_names[3]
- elif dom_models[i]==4:
- dom_models[i]=model_names[4]
- elif dom_models[i]==5:
- dom_models[i]=model_names[5]
- return dom_models
- def dominant(names, values):
- m = max(values)
- return [name for name, value in zip(names, values) if value == m]
- >>> model_names = ['is1', 'is5', 'is10', 'im1', 'im5', 'im10']
- >>> model_values = [0.1, 0.2, 0.1, 0.3, 0.2, 0.3]
- >>> dominant(model_names, model_values)
- ['im1', 'im10']
- def dominant_model(models):
- m = max(models)
- dom_models = [i for i,j in enumerate(models) if j==m]
- for i in range(len(dom_models)):
- for j in range(6):
- if dom_models[i]==j:
- dom_models[i]=model_names[j]
- return dom_models
Add Comment
Please, Sign In to add comment