Update generic.py

generic.py

@@ -39,7 +39,7 @@ def generate_visual(text_batch, target_class):
     token_importance = {}
     tokens = tokenizer.convert_ids_to_tokens(input_ids.flatten())
     for i in range(len(tokens)):
-        token_importance[tokens[i]] = expl[i].item()
+        token_importance[tokens[i]] = round(expl[i].item(), 3)
     vis_data_records = [visualization.VisualizationDataRecord(
         expl,
         output[0][classification],