Skip to content

Commit e49f860

Browse files
committed
update sigmoid
1 parent 54d41ea commit e49f860

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

text_classifier/prepare_dataset.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -704,9 +704,13 @@ def process_dataset(self, input_path: str, output_path: str = None, clean_text:
704704
print("ℹ️ No conversion needed - data is already suitable for sigmoid")
705705
else:
706706
print(f"⚠️ LLM advises against sigmoid: {activation_analysis.get('sigmoid_reasoning', 'N/A')}")
707-
print("💡 Falling back to softmax (multiclass) activation")
708-
analysis['final_task_type'] = 'multiclass'
709-
analysis['final_activation'] = 'softmax'
707+
print("🔧 However, user explicitly requested sigmoid - forcing multilabel conversion")
708+
# User explicitly requested sigmoid, so we force it even if LLM disagrees
709+
print("🔄 Converting to multilabel format (user override)...")
710+
df = self.convert_to_multilabel(df, analysis)
711+
analysis['final_task_type'] = 'multilabel'
712+
analysis['final_activation'] = 'sigmoid'
713+
print("✅ Forced conversion completed")
710714

711715
elif activation == 'softmax':
712716
print("\n⚡ User requested softmax activation (multiclass)")

0 commit comments

Comments
 (0)