CooperW's picture
Create README.md
18782e3 verified

This model has been trained for binary classification of the TON-Iot Network dataset.

Refer to adapter_config.json for LoRa Parameters.

Accuracy: 94.6 %

Key Training Params:

  • Steps: 100
  • Batch Size: 128
  • Optimizer: Adam_8bit
  • lr_schedular: cosine_with_restarts
  • lr_scheduler_kwargs = { "num_cycles": len(trainer.get_train_dataloader()) / 100 }

The datasets are an 80/20 partition of the TON-Iot Network dataset.

Train Dataset:

  • train_network.jsonl

Test Dataset:

  • test_network.jsonl