Advertisement
Guest User

Untitled

a guest
Dec 9th, 2018
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.89 KB | None | 0 0
  1. def get_arrage_status(name):
  2. model1 = CatBoostClassifier()
  3. model1.load_model("catboost_detect")
  4. a, num = add_data(name)
  5. img_data = np.array(a)
  6. img_data /= 255
  7. out_pred = model.predict_proba(img_data)
  8. s = ""
  9. uv = []
  10. out = ''
  11. for i in out_pred:
  12. #print(i)
  13. s += str(i.argmax())
  14. uv.append(i.max())
  15. k = len(s) // num
  16. #print(len(s))
  17. l = 0
  18. s += '9'
  19. r = 0
  20. c = '9'
  21. arr = []
  22. for i in s:
  23. if(i == '0'):
  24. c = '9'
  25. if i!=c:
  26. arr.append(l)
  27. c = i;
  28. l = 1
  29. else:
  30. l += 1
  31. arr.sort()
  32. arr.reverse()
  33. metric = np.array(arr[:5]).sum()
  34. return metric
  35. #print(metric)
  36. # for i in range(num):
  37. # for j in range(k):
  38. # out += s[r]
  39. # r+=1
  40. # out +='\n'
  41.  
  42. # print(out)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement