Aqsa Kausar commited on
Commit
3c3403a
·
unverified ·
1 Parent(s): f63b5f6

Update embedding_gen.py

Browse files
Files changed (1) hide show
  1. embedding_gen.py +1 -0
embedding_gen.py CHANGED
@@ -4,6 +4,7 @@ import numpy as np
4
  import umap
5
  import matplotlib.pyplot as plt
6
  import plotly.express as px
 
7
 
8
  # Step 1: Load skills from all files in a specific date folder
9
  def load_skills_from_date(base_folder, date):
 
4
  import umap
5
  import matplotlib.pyplot as plt
6
  import plotly.express as px
7
+ from sklearn.cluster import KMeans
8
 
9
  # Step 1: Load skills from all files in a specific date folder
10
  def load_skills_from_date(base_folder, date):