|
- import requests
- import json
- import time
- import os
-
- url = "http://192.168.202.130:8885/predict"
-
- langs_zh = ['vi', 'ko', 'en', 'nl',
- 'de', 'ms', 'id', 'tl',
- 'mn', 'my', 'th', 'lo',
- 'km', 'lt', 'et', 'lv',
- 'hu', 'pl', 'cs', 'sk',
- 'sl', 'hr', 'bs', 'sr',
- 'bg', 'mk', 'ru', 'uk',
- 'be', 'el', 'ka', 'hy',
- 'ro', 'fr', 'es', 'pt',
- 'fa', 'he', 'ar', 'ps',
- 'tr', 'kk', 'uz',
- 'az', 'hi', 'ta',
- 'ur', 'bn', 'ne']
-
- pt = r'result_Pangu'
-
-
- for lang in langs_zh:
-
- print('--------------'*10)
- print('translation start')
-
- src = "zh"
- tar = lang
- os.makedirs(pt + '/' + tar +'/')
- print('src:{} -> tar:{}'.format(src, tar))
-
- with open('test-silu/' + src + '.devtest','r',encoding='utf-8') as f:
- lines=f.readlines()
-
- f.close()
-
- times = []
- total_time = 0
- i = 0
-
- with open(pt + '/' + tar +'/' + src + '2' + tar + '.test','w') as f:
- for line in lines:
- i += 1
- data = json.dumps({
- "src_sentence": line,
- "src_langs_idx": src,
- "tar_langs_idx": tar
- })
- headers = {
- 'Content-Type': 'application/json'
- }
- starttime = time.time()
- response = requests.request("POST", url, headers=headers, data=data)
- endtime = time.time()
- response_time = endtime - starttime
- times.append(response_time)
- total_time += response_time
- f.writelines(response.text)
- f.writelines('\n')
- print(str(i) + '.' + response.text)
-
- f.close()
-
- everage_response_time = total_time / i
-
- with open(pt + '/' + tar +'/' + src + '2' + tar + '.txt','w') as f:
-
- for tm in times:
- f.writelines(str(tm))
- f.writelines('\n')
-
- f.writelines('total_time:'+str(total_time))
- f.writelines('\n')
- f.writelines('everage_response_time:'+str(everage_response_time))
-
- f.close()
- print('total_time')
- print(total_time)
- print('everage_response_time')
- print(everage_response_time)
-
-
- #------------------------------------------------测试外文翻译中文
-
- src = lang
- tar = "zh"
- print('src:{} -> tar:{}'.format(src, tar))
-
- with open('test-silu/' + src + '.devtest','r',encoding='utf-8') as f:
- lines=f.readlines()
-
- f.close()
-
- times = []
- total_time = 0
- i = 0
-
- with open(pt + '/' + src +'/' + src + '2' + tar + '.test','w') as f:
- for line in lines:
- i += 1
- data = json.dumps({
- "src_sentence": line,
- "src_langs_idx": src,
- "tar_langs_idx": tar
- })
- headers = {
- 'Content-Type': 'application/json'
- }
- starttime = time.time()
- response = requests.request("POST", url, headers=headers, data=data)
- endtime = time.time()
- response_time = endtime - starttime
- times.append(response_time)
- total_time += response_time
- f.writelines(response.text)
- f.writelines('\n')
- print(str(i) + '.' + response.text)
-
- f.close()
-
- everage_response_time = total_time / i
-
- with open(pt + '/' + src +'/' + src + '2' + tar + '.txt','w') as f:
-
- for tm in times:
- f.writelines(str(tm))
- f.writelines('\n')
-
- f.writelines('total_time:'+str(total_time))
- f.writelines('\n')
- f.writelines('everage_response_time:'+str(everage_response_time))
-
- f.close()
- print('total_time')
- print(total_time)
- print('everage_response_time')
- print(everage_response_time)
- print('translation completed')
|