derek-thomas HF staff commited on
Commit
9ae78a1
1 Parent(s): 08d3eb0

Updating score_percentiles column name

Browse files
Files changed (1) hide show
  1. src/build_nomic.py +1 -1
src/build_nomic.py CHANGED
@@ -28,7 +28,7 @@ def build_nomic(dataset):
28
 
29
  # Add a 'percentile_ranges' column to the DataFrame
30
  # This assigns each score to its corresponding percentile range
31
- df['percentile_ranges'] = pd.cut(df['score'], bins=bins, labels=labels, include_lowest=True)
32
 
33
  # Create Atlas project
34
  project = atlas.map_data(embeddings=np.stack(df['embedding'].values),
 
28
 
29
  # Add a 'percentile_ranges' column to the DataFrame
30
  # This assigns each score to its corresponding percentile range
31
+ df['score_percentile'] = pd.cut(df['score'], bins=bins, labels=labels, include_lowest=True)
32
 
33
  # Create Atlas project
34
  project = atlas.map_data(embeddings=np.stack(df['embedding'].values),