new model

app.py

@@ -19,6 +19,8 @@ def predict_by_model(Title, FullDescription, model):
 embedding_model = SentenceTransformer('all-mpnet-base-v2')
 with open("pickle_model_final.pkl", 'rb') as file:
     pickle_model = pickle.load(file)
+with open("pickle_model_try20.pkl", 'rb') as file:
+    pickle_model20 = pickle.load(file)
 
 
 categories = ['', 
@@ -52,25 +54,53 @@ categories = ['',
      'Trade & Construction Jobs',
      'Travel Jobs']
      
-def GetResult(description, text, model):
+def GetResultV(description, text, model):
     prediction = predict_by_model(description, text, model)
     return int((prediction // 500) * 500)
+    
+    
+def GetResultN(title, description, category, contractType, contractTime, model):
+        d = {'Title': [normilize_text(title)], 
+            'FullDescription': [normilize_text(description)],
+            'ContractTime'+'_contract': [contractTime == 'contract'],
+            'ContractTime'+'_permanent': [contractTime == 'permanent'],
+            'ContractTime'+'_unknown': [contractTime != 'contract' and contractTime != 'permanent'],
+            'ContractType'+'_full_time': [contractType == 'full_time'],
+            'ContractType'+'_part_time': [contractType == 'part_time'],
+            'ContractType'+'_unknown': [contractType != 'full_time' and contractType != 'part_time'],
+        }
+    for i in Category.keys():
+      d[i] = 0
+    d[category] = 1
+    Valid = pd.DataFrame(d)
+    title_emb, descr_emb = pd_to_emb_pd(Valid)
+    Valid = Valid.join(pd.DataFrame(title_emb, columns=title123)).join(pd.DataFrame(descr_emb, columns=descr123))
+    del Valid['Title']
+    del Valid['FullDescription']
+
+    predict = model.predict(Valid)
+    return int((predict // 500) * 500)
      
 
 st.warning('Введите название и описание вакансии')    
 
 description = st.text_input('Введите название вакансии')
-text = st.text_area('Введите описание вакансии')
-
-option = st.selectbox('Введите категорию вакансии', categories)
+title = st.text_area('Введите описание вакансии')
+category = st.selectbox('Введите категорию вакансии', categories)
+contractType = st.selectbox('Введите тип занятости', ['part_time', 'full_time', 'other'])
+contractTime = st.selectbox('Введите тип договора', ['contract', 'permanent', 'other'])
 
 if st.button('Предсказать зарплату'):
-    if description != '' and text != '':
-        result = GetResult(description, text, pickle_model)
+    if description != '' and title != '':
+        if len(description) < 50 or len(title) < 15:
+            st.error('Поле с названием и описанием вакансии имеют слишком короткие значения')
+            return
+        #result = GetResult(description, title, pickle_model)
+        result = GetResultN(title, description, category, contractType, contractTime, pickle_model20)
         st.write('Результат: ' + str(result))
     else:
         st.error('Поле с названием и описанием вакансии должны быть обязательно заполнены')
 else:
     st.write('Результат:')
-    if description != '' and text != '':
+    if description != '' and title != '':
         st.warning('Для получения предсказания нажмите кнопку "Предсказать зарплату"')