| { | |
| "eval": { | |
| "loss": 3.249854749848661e-05, | |
| "accuracy": 1.0, | |
| "samples": 360, | |
| "confusion": { | |
| "general_agent": { | |
| "general_agent": 120, | |
| "ckan_retrieval": 0, | |
| "openui_translator": 0 | |
| }, | |
| "ckan_retrieval": { | |
| "general_agent": 0, | |
| "ckan_retrieval": 120, | |
| "openui_translator": 0 | |
| }, | |
| "openui_translator": { | |
| "general_agent": 0, | |
| "ckan_retrieval": 0, | |
| "openui_translator": 120 | |
| } | |
| } | |
| }, | |
| "test": { | |
| "loss": 9.822394076763885e-05, | |
| "accuracy": 1.0, | |
| "samples": 360, | |
| "confusion": { | |
| "general_agent": { | |
| "general_agent": 120, | |
| "ckan_retrieval": 0, | |
| "openui_translator": 0 | |
| }, | |
| "ckan_retrieval": { | |
| "general_agent": 0, | |
| "ckan_retrieval": 120, | |
| "openui_translator": 0 | |
| }, | |
| "openui_translator": { | |
| "general_agent": 0, | |
| "ckan_retrieval": 0, | |
| "openui_translator": 120 | |
| } | |
| } | |
| }, | |
| "history": [ | |
| { | |
| "epoch": 1, | |
| "train_loss": 0.02816176313383115, | |
| "eval_loss": 0.00021523007654751484, | |
| "eval_accuracy": 1.0 | |
| }, | |
| { | |
| "epoch": 2, | |
| "train_loss": 5.643083297507524e-05, | |
| "eval_loss": 3.249854749848661e-05, | |
| "eval_accuracy": 1.0 | |
| } | |
| ], | |
| "config": { | |
| "vocab_size": 130072, | |
| "embedding_dim": null, | |
| "hidden_dim": 128, | |
| "num_labels": 3, | |
| "dropout": 0.1, | |
| "pad_token_id": 1, | |
| "encoder_model_name": "openbmb/MiniCPM5-1B", | |
| "encoder_hidden_size": 1536, | |
| "architecture": "frozen_encoder", | |
| "labels": [ | |
| "general_agent", | |
| "ckan_retrieval", | |
| "openui_translator" | |
| ] | |
| } | |
| } | |