AbstractIntegratedModule 1.2.1__tar.gz → 1.2.2__tar.gz

{abstractintegratedmodule-1.2.1 → abstractintegratedmodule-1.2.2}/AbstractIntegratedModule.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AbstractIntegratedModule
-Version: 1.2.1
+Version: 1.2.2
 Summary: Integrated Pipeline with Specialized Non-LLM AI Agent Framework for ARM64 architecture
 Author: Micro-Novelty
 Author-email: Micro-Novelty <hernikpuspita5@gmail.com>

{abstractintegratedmodule-1.2.1 → abstractintegratedmodule-1.2.2}/AbstractIntegratedModule.py

@@ -1923,7 +1923,7 @@ class Transformer:
             mask_expanded = np.broadcast_to(
                 mask, (batch, heads, seq_len, seq_len)
             )
-            # FIX 1 — inplace operations, ARM64 NEON works better on contiguous memory
+            # inplace operations, ARM64 NEON works better on contiguous memory
             attn_weights = attn_weights * mask_expanded
             row_sums     = attn_weights.sum(axis=-1, keepdims=True) + eps
             attn_weights = attn_weights / row_sums
@@ -2727,7 +2727,6 @@ class AdamOptimizer:
 class MLP:
     def __init__(self):
         self.layers = []
-        self.layers2 = []
         self.lr = 0.1
 
         self.error_counts = None
@@ -17591,11 +17590,13 @@ class PipelinePredictionManager:
 
         return titles, y, label_map
 
-    def knn_forward_inference(self, X, y, memory_metric='euclidean', training=False, batch_size=2, train_mode='fixed_backward', lr=0.1):
+    def knn_forward_inference(self, X, y, label_map=None, memory_metric='euclidean', training=False, batch_size=2, train_mode='fixed_backward', lr=0.1):
         trans_probs = None
         use_embedded = True
         num_classes = self.pipeline._get_num_classes()
-
+        if label_map and isinstance(label_map, dict):
+            reverse_map = {v: k for k, v in label_map.items()}
+    
         if self.pipeline.model2 is not None:
             self.pipeline.knn_tf = KNNAugmentedTransformer(
                                  vocab_size=self.pipeline.vocab_size, d_model=self.pipeline.transformer_d_model, n_heads=self.pipeline.transformer_heads, num_classes=num_classes,
@@ -17652,19 +17653,28 @@ class PipelinePredictionManager:
                 print('[>] Starting kNN Augmented Transformer Training...')
                 sequence_ids_train = self.pipeline._features_to_sequence(X_train)
                 losses, accs = self.pipeline.knn_tf.train(input_ids_list=sequence_ids_train, y_true_list=y, epochs=self.pipeline.transformer_training_epochs, mode=train_mode, lr=lr, embedded=True, batch_size=batch_size)
-                return losses, accs
+                return losses, accs, None
             else:
-                print('[>] Returning Transformer probabilities directly...')
+                print('[>] Processing kNN Augmented Transformer probabilities...')
                 sequence_ids = self.pipeline._features_to_sequence(X)
                 trans_probs, attn_weights = self.pipeline.knn_tf.forward(input_ids=sequence_ids, AME=AME, embedded=use_embedded)
 
-            return trans_probs, attn_weights
+            if label_map is None:
+                print('[⚠️] > label_map is None, returning kNN Transformer probabilities without any label mapping...')
+                return trans_probs, attn_weights, None
+            else:
+                trans_pred_indices = np.argmax(trans_probs, axis=-1).tolist()
+                trans_pred_idx = max(set(trans_pred_indices), key=trans_pred_indices.count)
+                predict_confidence = trans_probs[0][trans_pred_idx]
+                prediction = reverse_map.get(trans_pred_idx, 'Unknown')
+                print(f"[=] kNN Augmented Transformer Prediction: {prediction} with probability {predict_confidence:.2%}")
+                return trans_probs, attn_weights, prediction
         
         except Exception as e:
             print(f'[>] Error initating kNN forward Inference: {e}')
             traceback.print_exc()
-            return [], []
-        
+            return [], [], None
+
 
 
     def regular_prediction_method(self, titles=None, label_map=None, rules=None, X=None, y=None, show_proba=False, top_k=3, batch_size=2, use_transformer=True):
@@ -19304,7 +19314,7 @@ class PipelinePredictionManager:
                         print(f"\n[⚠️] Final prediction is {chosen_label} with uncertain confidence: {confidence:.1%}. Consider more consistent data for the model to learn from.")
                     else:
                         print(f"\n[🎯] Predicted label: {chosen_label} || With Certain Confidence: {confidence:.1%}")
-                    
+
                 return results, chosen_label, confidence
 
             else:
@@ -19652,7 +19662,7 @@ class PipelinePredictionManager:
                 
         except Exception as e:
             print(f'[!] Cant check and calibrate probs based on penalty due to: {e}') 
-
+            
         return final_probs