0xOracle's picture
added images and updated model card description
94596fb
metadata
license: apache-2.0
base_model: Intel/neural-chat-7b-v3-1
tags:
  - generated_from_trainer
model-index:
  - name: neural-chat-finetuned-bilic-v1
    results: []

LLM_IMAGE

neural-chat-finetuned-bilic-v1

This model is a fine-tuned version of Intel/neural-chat-7b-v3-1 on our custom dataset.

Model description

This is a fine tuned version of the intel's Neuralchat model, specifically trained on a carefully curated dataset on fraud detection. We implemented a contextual based architecture to enable the model learn and be adept at understanding context within a conversation as opposed to the traditional rule based approach.

Intended uses & limitations

  • detecting fraudulent conversations in real-time
  • Giving a summary of conversations and suggestions
  • Understanding with high accuracy the context in a conversation to make better predictions

Training

50,000 synthetically conversations

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.0002
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • training_steps: 250
  • mixed_precision_training: Native AMP

Framework versions

  • Transformers 4.36.0.dev0
  • Pytorch 2.1.0+cu118
  • Datasets 2.15.0
  • Tokenizers 0.15.0