Skip to content

Commit

Permalink
remove model and text column from output
Browse files Browse the repository at this point in the history
  • Loading branch information
Thomas-S-Allen committed Mar 14, 2024
1 parent b283bac commit ec2b251
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion quick_classifier.py
Expand Up @@ -90,7 +90,6 @@
# Start with just bibcodes
bibcodes = records['bibcode'].tolist()


# Harvest Title and Abstract from Solr
records = harvest_solr(bibcodes, start_index=0, fields='bibcode, title, abstract')

Expand All @@ -114,6 +113,11 @@
# Update records
record['collection_scores'] = collection_scores
record['minimum_collection_score'] = min(collection_scores)
# Remove "model" and "text keys from record
del record['model']
del record['text']

# Overwrite record with updated record
records[index] = record


Expand Down

0 comments on commit ec2b251

Please sign in to comment.