Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def get_arrage_status(name):
- model1 = CatBoostClassifier()
- model1.load_model("catboost_detect")
- a, num = add_data(name)
- img_data = np.array(a)
- img_data /= 255
- out_pred = model.predict_proba(img_data)
- s = ""
- uv = []
- out = ''
- for i in out_pred:
- #print(i)
- s += str(i.argmax())
- uv.append(i.max())
- k = len(s) // num
- #print(len(s))
- l = 0
- s += '9'
- r = 0
- c = '9'
- arr = []
- for i in s:
- if(i == '0'):
- c = '9'
- if i!=c:
- arr.append(l)
- c = i;
- l = 1
- else:
- l += 1
- arr.sort()
- arr.reverse()
- metric = np.array(arr[:5]).sum()
- return metric
- #print(metric)
- # for i in range(num):
- # for j in range(k):
- # out += s[r]
- # r+=1
- # out +='\n'
- # print(out)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement