Skip to content

Commit

Permalink
add some explanation for printed things
Browse files Browse the repository at this point in the history
  • Loading branch information
ericliu8168 committed Aug 9, 2024
1 parent 4d81aad commit 72b663a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions libmultilabel/linear/tree.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,10 +147,10 @@ def count(node):

# Calculate the total memory (excluding swap) on the local machine
total_memory = psutil.virtual_memory().total
print(f'{total_memory / (1024**3):.3f} GB')
print(f'Your system memory is: {total_memory / (1024**3):.3f} GB')

model_size = get_estimated_model_size(root, num_nodes)
print(f'*** model_size: {model_size / (1024**3):.3f} GB')
print(f'The estimated tree model size is: {model_size / (1024**3):.3f} GB')

if (total_memory <= model_size):
raise MemoryError(f'Not enough memory to train the model. model_size: {model_size / (1024**3):.3f} GB')
Expand Down

0 comments on commit 72b663a

Please sign in to comment.