{ "cells": [ { "cell_type": "code", "execution_count": 8, "id": "9dd681f2", "metadata": {}, "outputs": [], "source": [ "import pandas as pd\n", "import matplotlib.pyplot as plt\n", "import ast\n", "import json" ] }, { "cell_type": "code", "execution_count": 2, "id": "badd029d", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
| \n", " | number | \n", "duration_sec | \n", "train_loss | \n", "val_loss | \n", "batch_size | \n", "lr | \n", "weight_decay | \n", "contrastive_temp | \n", "peak_dropout | \n", "formula_attn_heads | \n", "formula_transformer_layers | \n", "formula_dims | \n", "gnn_dropout | \n", "gnn_channels | \n", "final_embedding_dim | \n", "
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0 | \n", "0 | \n", "24041.086241 | \n", "[3.6679911613464355, 3.209094285964966, 3.0371... | \n", "[4.160371780395508, 4.162205696105957, 3.68366... | \n", "64 | \n", "0.000045 | \n", "0.000080 | \n", "0.086119 | \n", "0.235952 | \n", "4 | \n", "4 | \n", "256,512 | \n", "0.463260 | \n", "256,512 | \n", "256 | \n", "
| 1 | \n", "1 | \n", "33999.050235 | \n", "[4.855160713195801, 4.80089807510376, 4.646565... | \n", "[4.843686103820801, 4.86243200302124, 4.816713... | \n", "128 | \n", "0.000004 | \n", "0.000002 | \n", "0.085358 | \n", "0.496695 | \n", "2 | \n", "2 | \n", "512,256 | \n", "0.246349 | \n", "64,128,128 | \n", "512 | \n", "
| 2 | \n", "2 | \n", "24040.417972 | \n", "[4.1822099685668945, 4.158326148986816, 4.1557... | \n", "[4.1694536209106445, 4.173160552978516, 4.1776... | \n", "64 | \n", "0.000623 | \n", "0.000317 | \n", "0.057541 | \n", "0.479595 | \n", "2 | \n", "4 | \n", "64,128 | \n", "0.467235 | \n", "256,512 | \n", "256 | \n", "
| 3 | \n", "3 | \n", "20089.766765 | \n", "[3.6783478260040283, 3.126972198486328, 2.9364... | \n", "[4.160656452178955, 4.162418365478516, 3.66769... | \n", "64 | \n", "0.000026 | \n", "0.009015 | \n", "0.058374 | \n", "0.379269 | \n", "4 | \n", "2 | \n", "512,256 | \n", "0.130730 | \n", "128,256 | \n", "256 | \n", "
| 4 | \n", "4 | \n", "26225.759020 | \n", "[3.7406771183013916, 3.1643409729003906, 2.951... | \n", "[4.180196285247803, 4.194350719451904, 3.78971... | \n", "64 | \n", "0.000029 | \n", "0.000012 | \n", "0.022773 | \n", "0.414426 | \n", "4 | \n", "2 | \n", "512,256 | \n", "0.232350 | \n", "64,128 | \n", "512 | \n", "