cat-llm 0.0.71__tar.gz → 0.0.72__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cat-llm
3
- Version: 0.0.71
3
+ Version: 0.0.72
4
4
  Summary: A tool for categorizing text data and images using LLMs and vision models
5
5
  Project-URL: Documentation, https://github.com/chrissoria/cat-llm#readme
6
6
  Project-URL: Issues, https://github.com/chrissoria/cat-llm/issues
@@ -1,7 +1,7 @@
1
1
  # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
2
2
  #
3
3
  # SPDX-License-Identifier: MIT
4
- __version__ = "0.0.71"
4
+ __version__ = "0.0.72"
5
5
  __author__ = "Chris Soria"
6
6
  __email__ = "chrissoria@berkeley.edu"
7
7
  __title__ = "cat-llm"
@@ -733,11 +733,11 @@ def multi_class(
733
733
 
734
734
  categorized_data[cat_cols] = categorized_data[cat_cols].astype('Int64')
735
735
 
736
- categorized_data['categories_present'] = categorized_data[cat_cols].apply(
736
+ categorized_data['categories_id'] = categorized_data[cat_cols].apply(
737
737
  lambda x: ','.join(x.dropna().astype(str)), axis=1
738
738
  )
739
739
 
740
- categorized_data['categories_counted'] = categorized_data[cat_cols].count(axis=1)
740
+ categorized_data['categories_present'] = categorized_data[cat_cols].sum(axis=1)
741
741
 
742
742
  if to_csv:
743
743
  if save_directory is None:
File without changes
File without changes
File without changes
File without changes