Skip to content
Snippets Groups Projects
Commit c94738ad authored by deep1401's avatar deep1401
Browse files

Add tensorboard to evals

parent 8f860090
No related branches found
No related tags found
No related merge requests found
...@@ -16,11 +16,13 @@ import { ...@@ -16,11 +16,13 @@ import {
FileDigitIcon, FileDigitIcon,
Grid3X3Icon, Grid3X3Icon,
Trash2Icon, Trash2Icon,
LineChartIcon,
Type, Type,
} from 'lucide-react'; } from 'lucide-react';
import { useState, useEffect } from 'react'; import { useState, useEffect } from 'react';
import useSWR from 'swr'; import useSWR from 'swr';
import * as chatAPI from '../../../lib/transformerlab-api-sdk'; import * as chatAPI from '../../../lib/transformerlab-api-sdk';
import TensorboardModal from '../Train/TensorboardModal';
import ViewOutputModalStreaming from './ViewOutputModalStreaming'; import ViewOutputModalStreaming from './ViewOutputModalStreaming';
import ViewCSVModal from './ViewCSVModal'; import ViewCSVModal from './ViewCSVModal';
import ViewPlotModal from './ViewPlotModal'; import ViewPlotModal from './ViewPlotModal';
...@@ -96,8 +98,8 @@ const EvalJobsTable = () => { ...@@ -96,8 +98,8 @@ const EvalJobsTable = () => {
const [openPlotModal, setOpenPlotModal] = useState(false); const [openPlotModal, setOpenPlotModal] = useState(false);
const [currentJobId, setCurrentJobId] = useState(''); const [currentJobId, setCurrentJobId] = useState('');
const [currentScore, setCurrentScore] = useState(''); const [currentScore, setCurrentScore] = useState('');
const [fileNameForDetailedReport, setFileNameForDetailedReport] = const [currentTensorboardForModal, setCurrentTensorboardForModal] = useState(-1);
useState(''); const [fileNameForDetailedReport, setFileNameForDetailedReport] = useState('');
const fetchCSV = async (jobId) => { const fetchCSV = async (jobId) => {
const response = await fetch( const response = await fetch(
...@@ -151,6 +153,10 @@ const EvalJobsTable = () => { ...@@ -151,6 +153,10 @@ const EvalJobsTable = () => {
setFileName={setFileNameForDetailedReport} setFileName={setFileNameForDetailedReport}
fileName={fileNameForDetailedReport} fileName={fileNameForDetailedReport}
/> />
<TensorboardModal
currentTensorboard={currentTensorboardForModal}
setCurrentTensorboard={setCurrentTensorboardForModal}
/>
<Box <Box
sx={{ sx={{
display: 'flex', display: 'flex',
...@@ -172,6 +178,8 @@ const EvalJobsTable = () => { ...@@ -172,6 +178,8 @@ const EvalJobsTable = () => {
</Typography> </Typography>
)} )}
</Box> </Box>
<Typography level="h3">Executions</Typography>
<Sheet sx={{ overflowY: 'scroll' }}> <Sheet sx={{ overflowY: 'scroll' }}>
<Table stickyHeader> <Table stickyHeader>
<thead> <thead>
...@@ -285,6 +293,18 @@ const EvalJobsTable = () => { ...@@ -285,6 +293,18 @@ const EvalJobsTable = () => {
variant="soft" variant="soft"
sx={{ justifyContent: 'flex-end' }} sx={{ justifyContent: 'flex-end' }}
> >
{job?.job_data?.tensorboard_output_dir && (
<Button
size="sm"
variant="plain"
onClick={() => {
setCurrentTensorboardForModal(job?.id);
}}
startDecorator={<LineChartIcon />}
>
Tensorboard
</Button>
)}
<Button <Button
onClick={() => { onClick={() => {
setViewOutputFromJob(job?.id); setViewOutputFromJob(job?.id);
......
...@@ -287,7 +287,7 @@ export default function EvalModal({ ...@@ -287,7 +287,7 @@ export default function EvalModal({
} else { } else {
console.log('formJson:', formJson); console.log('formJson:', formJson);
const template_name = formJson.template_name; const template_name = formJson.template_name;
delete formJson.template_name; // delete formJson.template_name;
const result = await chatAPI.EXPERIMENT_ADD_EVALUATION( const result = await chatAPI.EXPERIMENT_ADD_EVALUATION(
experimentInfo?.id, experimentInfo?.id,
template_name, template_name,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment