Skip to content

Commit

Permalink
feat(modeldetails): adding properties for supporting case classificat…
Browse files Browse the repository at this point in the history
…ion (#902)
  • Loading branch information
lprovost-coveo authored Jan 8, 2025
1 parent 9571063 commit 47cf59c
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import {ModelDetailsBuildingStats} from './ModelDetailsBuildingStats.js';
import {ModelDetailsLanguages} from './ModelDetailsLanguages.js';
import {ModelDetailsSubModels} from './ModelDetailsSubModels.js';
import {ModelDetailsPreparationDetails} from './ModelDetailsPreparationDetails.js';
import {ModelDetailsTrainingDetails} from './ModelDetailsTrainingDetails.js';

export interface ModelDetails {
possibleRecommendations?: number;
Expand All @@ -17,4 +19,6 @@ export interface ModelDetails {
subModels?: {[key: string]: ModelDetailsSubModels};
candidates?: number;
modelDetailedBuildingStats?: ModelDetailsBuildingStats;
modelDetailedTrainingDetails: ModelDetailsTrainingDetails;
modelDetailedPreparationDetails: ModelDetailsPreparationDetails;
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
interface DatasetFieldDetails {
numSamples: number;
labelsDistribution: Record<string, number>;
}

interface FacetDetails {
facetLabels: Record<string, string[]>;
}

interface DatasetDetails {
numRows: number;
dataDetails: Record<string, DatasetFieldDetails>;
}

export interface ModelDetailsPreparationDetails {
trainDatasetsDetails: DatasetDetails;
testDatasetsDetails: DatasetDetails;
facetDetails: FacetDetails;
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
interface FacetPerformanceDetails {
hit1: number;
hit3: number;
}

export interface ModelDetailsTrainingDetails {
performanceDetails: Record<string, FacetPerformanceDetails>;
}

0 comments on commit 47cf59c

Please sign in to comment.