TechNeur/course_search

1import pickle2import pandas as pd3from sentence_transformers import SentenceTransformer4from sklearn.metrics.pairwise import cosine_similarity5 6with open("course_emb.pkl", "rb") as f:7    course_emb = pickle.load(f)8 9df = pd.read_excel("analytics_vidhya_courses_Final.xlsx")10 11model = SentenceTransformer('all-MiniLM-L6-v2')12 13def search_courses(query, top_n=5):14    query_embedding = model.encode([query])15    16    similarities = cosine_similarity(query_embedding, course_emb)17    18    top_n_idx = similarities[0].argsort()[-top_n:][::-1]19    20    return df.iloc[top_n_idx][["Course Title", "Course Description"]]21 22query = input("Enter your search query: ")23top_courses = search_courses(query)24 25print("\nTop relevant courses:")26for idx, row in top_courses.iterrows():27    print(f"Title: {row['Course Title']}")28    print(f"Description: {row['Course Description']}\n")29