forked from mrlan/EnglishPal
				
			
							parent
							
								
									c4378e73cd
								
							
						
					
					
						commit
						a83f91a88d
					
				|  | @ -24,7 +24,7 @@ def convert_test_type_to_difficulty_level(d): | ||||||
|     :return: |     :return: | ||||||
|     """ |     """ | ||||||
|     result = {} |     result = {} | ||||||
|     L = list(d.keys())  # in dic, we have test types (e.g., CET4,CET6,BBC) for each word |     L = list(d.keys())  # in d, we have test types (e.g., CET4,CET6,BBC) for each word | ||||||
| 
 | 
 | ||||||
|     for k in L: |     for k in L: | ||||||
|         if 'CET4' in d[k]: |         if 'CET4' in d[k]: | ||||||
|  | @ -33,30 +33,13 @@ def convert_test_type_to_difficulty_level(d): | ||||||
|             result[k] = 5 |             result[k] = 5 | ||||||
|         elif 'CET6' in d[k] or 'GRADUATE' in d[k]: |         elif 'CET6' in d[k] or 'GRADUATE' in d[k]: | ||||||
|             result[k] = 6 |             result[k] = 6 | ||||||
|         elif 'OXFORD5000' in d[k]: |         elif 'OXFORD5000' in d[k] or 'IELTS' in d[k]: | ||||||
|             result[k] = 7 |             result[k] = 7 | ||||||
|         elif 'BBC' in d[k]: |         elif 'BBC' in d[k]: | ||||||
|             result[k] = 8 |             result[k] = 8 | ||||||
| 
 | 
 | ||||||
|     return result  # {'apple': 4, ...} |     return result  # {'apple': 4, ...} | ||||||
| 
 | 
 | ||||||
| def simplify_the_words_dict(d): |  | ||||||
|     """ |  | ||||||
|     用于把保存了词库中评级后的词新建一个以词根为键、以同词根的最低等级为值 |  | ||||||
|     """ |  | ||||||
|     stem = snowballstemmer.stemmer('english') |  | ||||||
| 
 |  | ||||||
|     result = {} |  | ||||||
|     for k in d:   # j 在字典中 |  | ||||||
|         temp = stem.stemWord(k)     # 提取j得词根 |  | ||||||
|         if temp not in result:         # 如果这个词根不在结果字典中,则以词根为键、以dic中的等级作为值添加 |  | ||||||
|             result[temp] = d[k] |  | ||||||
|         else:                   # 如果这个词在结果词典中,则比较一下单词的难度等级是否最小 |  | ||||||
|             if result[temp] > d[k]: |  | ||||||
|                 result[temp] = d[k] |  | ||||||
| 
 |  | ||||||
|     return result |  | ||||||
| 
 |  | ||||||
| 
 | 
 | ||||||
| def get_difficulty_level_for_user(d1, d2): | def get_difficulty_level_for_user(d1, d2): | ||||||
|     """ |     """ | ||||||
|  | @ -65,14 +48,13 @@ def get_difficulty_level_for_user(d1, d2): | ||||||
|     在d2的后面添加单词,没有新建一个新的字典 |     在d2的后面添加单词,没有新建一个新的字典 | ||||||
|     """ |     """ | ||||||
|     d2 = convert_test_type_to_difficulty_level(d2)   # 根据d2的标记评级{'apple': 4, 'abandon': 4, ...} |     d2 = convert_test_type_to_difficulty_level(d2)   # 根据d2的标记评级{'apple': 4, 'abandon': 4, ...} | ||||||
|     d2_simplified = simplify_the_words_dict(d2)                # 提取d2的词根   {'appl': 4, 'abandon': 4, ...} |  | ||||||
|     stem = snowballstemmer.stemmer('english') |     stem = snowballstemmer.stemmer('english') | ||||||
| 
 | 
 | ||||||
|     for k in d1:        # 用户的词 |     for k in d1:        # 用户的词 | ||||||
|         if k in d2:     # 如果用户的词以原型的形式存在于词库d2中 |         if k in d2:     # 如果用户的词以原型的形式存在于词库d2中 | ||||||
|             continue    # 无需评级,跳过 |             continue    # 无需评级,跳过 | ||||||
|         elif stem.stemWord(k) in d2_simplified:     # 如果用户的词的词根存在于词库d2的词根库中 |         elif stem.stemWord(k) in d2:     # 如果用户的词的词根存在于词库d2的词根库中 | ||||||
|             d2[k] = d2_simplified[k]        # 按照词根进行评级 |             d2[k] = d2[stem.stemWord(k)]        # 按照词根进行评级 | ||||||
|             break |             break | ||||||
|         else: |         else: | ||||||
|             d2[k] = 3       # 如果k的词根都不在,那么就当认为是3级 |             d2[k] = 3       # 如果k的词根都不在,那么就当认为是3级 | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue