perezcatriel commited on
Commit
5606168
·
1 Parent(s): 2c5c5bd
Files changed (2) hide show
  1. ML/skill_top.py +4 -4
  2. app.py +4 -2
ML/skill_top.py CHANGED
@@ -27,12 +27,12 @@ st.title("Predicción de aptitud para un trabajo")
27
  st.write(
28
  "Ingrese el título del trabajo para ver las habilidades más importantes.")
29
 
30
- title = st.multiselect("Título del trabajo", df.job_title.unique())
31
 
32
  # Crear una función que encuentra las habilidades más importantes para un título dado
33
  def get_top_skills(title, limit):
34
  # Filtrar el dataframe por el título dado
35
- filtered_df = df[df.job_title == title]
36
 
37
  # Crear una matriz de características con la frecuencia de cada skill en el dataframe filtrado
38
  X_filtered = vectorizer.transform(filtered_df.skills)
@@ -53,7 +53,7 @@ def get_top_skills(title, limit):
53
 
54
  if title:
55
  limit = st.number_input("Cantidad de habilidades a mostrar", value=5, min_value=1, max_value=len(all_skills))
56
- top_skills = get_top_skills(title[0], limit)
57
- st.write(f"Las {limit} habilidades más importantes para el trabajo de '{title[0]}' son:")
58
  for skill in top_skills:
59
  st.write(f"- {skill}")
 
27
  st.write(
28
  "Ingrese el título del trabajo para ver las habilidades más importantes.")
29
 
30
+ title = st.text_input("Título del trabajo")
31
 
32
  # Crear una función que encuentra las habilidades más importantes para un título dado
33
  def get_top_skills(title, limit):
34
  # Filtrar el dataframe por el título dado
35
+ filtered_df = df[df.job_title.str.contains(title, case=False)]
36
 
37
  # Crear una matriz de características con la frecuencia de cada skill en el dataframe filtrado
38
  X_filtered = vectorizer.transform(filtered_df.skills)
 
53
 
54
  if title:
55
  limit = st.number_input("Cantidad de habilidades a mostrar", value=5, min_value=1, max_value=len(all_skills))
56
+ top_skills = get_top_skills(title, limit)
57
+ st.write(f"Las {limit} habilidades más importantes para el trabajo de '{title}' son:")
58
  for skill in top_skills:
59
  st.write(f"- {skill}")
app.py CHANGED
@@ -435,9 +435,11 @@ if selected2 == "New":
435
  clf.fit(X, df.Aptitude)
436
 
437
  # Crear la interfaz de usuario con Streamlit
438
- st.title("Predicción de aptitud para un trabajo")
439
  st.write(
440
- "Ingrese el título del trabajo para ver las habilidades más importantes.")
 
 
441
 
442
  title = st.multiselect("Título del trabajo", df.job_title.unique())
443
 
 
435
  clf.fit(X, df.Aptitude)
436
 
437
  # Crear la interfaz de usuario con Streamlit
438
+ st.title("Predicción de aptitud para un 'NUEVO' trabajo")
439
  st.write(
440
+ "Ingrese el título del 'Nuevo' trabajo para ver que habilidades "
441
+ "son las más "
442
+ "importantes.")
443
 
444
  title = st.multiselect("Título del trabajo", df.job_title.unique())
445