public Experiment BuildEntity(IMLOpsDbContext db, Experiment entity)
{
if (entity.Runs == null)
{
db.Entry(entity).Collection(x => x.Runs).Load();
entity.Runs.ForEach(run => this.runBuilder.BuildEntity(db, run));
}
return(entity);
}
public Run BuildEntity(IMLOpsDbContext db, Run entity)
{
db.Entry(entity).Collection(x => x.HyperParameters).Load();
db.Entry(entity).Collection(x => x.Metrics).Load();
db.Entry(entity).Reference(x => x.ConfusionMatrix).Load();
db.Entry(entity).Collection(x => x.RunArtifacts).Load();
return(entity);
}
private void PopulateRun(IMLOpsDbContext db, Run run)
{
if (run == null)
{
return;
}
run.HyperParameters = db.HyperParameters.Where(x => x.RunId == run.RunId).ToList();
run.Metrics = db.Metrics.Where(x => x.RunId == run.RunId).ToList();
run.ConfusionMatrix = db.ConfusionMatrices.FirstOrDefault(x => x.RunId == run.RunId);
}
public Data BuildEntity(IMLOpsDbContext db, Data entity)
{
db.Entry(entity).Reference(x => x.DataSchema).Load();
db.Entry(entity.DataSchema).Collection(x => x.DataColumns).Load();
foreach (var dataColumn in entity.DataSchema.DataColumns)
{
db.Entry(dataColumn).Collection(x => x.DataDistributions).Load();
}
return(entity);
}
protected async Task TearDown(IMLOpsDbContext context)
{
var experiments = context.Experiments;
var runs = context.Runs;
var metrics = context.Metrics;
var hyperParameters = context.HyperParameters;
var confusionMatrices = context.ConfusionMatrices;
var data = context.Data;
var runArtifacts = context.RunArtifacts;
var registeredModels = context.RegisteredModels;
context.Experiments.RemoveRange(experiments);
context.Runs.RemoveRange(runs);
context.Metrics.RemoveRange(metrics);
context.HyperParameters.RemoveRange(hyperParameters);
context.ConfusionMatrices.RemoveRange(confusionMatrices);
context.Data.RemoveRange(data);
context.RunArtifacts.RemoveRange(runArtifacts);
context.RegisteredModels.RemoveRange(registeredModels);
await context.SaveChangesAsync();
}
protected async Task TearDown(IMLOpsDbContext context)
{
context.Experiments.RemoveRange(context.Experiments);
await context.SaveChangesAsync();
}
private void PopulateRegisteredModel(IMLOpsDbContext db, RegisteredModel registeredModel)
{
registeredModel.RunArtifact = db.RunArtifacts.First(x => x.RunArtifactId == registeredModel.RunArtifactId);
registeredModel.Run = db.Runs.First(x => x.RunId == registeredModel.RunId);
registeredModel.Experiment = db.Experiments.First(x => x.ExperimentId == registeredModel.ExperimentId);
}
public RegisteredModel BuildEntity(IMLOpsDbContext db, RegisteredModel entity)
{
db.Entry(entity).Collection(x => x.Deployments).Load();
return(entity);
}
public List <RegisteredModel> BuildEntities(IMLOpsDbContext db, List <RegisteredModel> entities)
{
return(entities.Select(x => BuildEntity(db, x)).ToList());
}
public List <Experiment> BuildEntities(IMLOpsDbContext db, List <Experiment> entities)
{
return(entities.Select(x => BuildEntity(db, x)).ToList());
}
public DeploymentTarget BuildEntity(IMLOpsDbContext db, DeploymentTarget entity)
{
db.Entry(entity).Collection(x => x.Deployments).Load();
return(entity);
}
public List <DeploymentTarget> BuildEntities(IMLOpsDbContext db, List <DeploymentTarget> entities)
{
return(entities.Select(x => BuildEntity(db, x)).ToList());
}
