Browse Source

added submit param not found funcs

pull/339/head
tzwang 1 year ago
parent
commit
6f83faea04
1 changed files with 17 additions and 0 deletions
  1. +17
    -0
      internal/storeLink/storeLink.go

+ 17
- 0
internal/storeLink/storeLink.go View File

@@ -73,6 +73,11 @@ const (
) )


var ( var (
Datasets = []string{"mnist", "cifar10"}
AlgorithmsForDatasets = map[string][]string{
"mnist": {"fcn"},
"cifar10": {"cnn"},
}
OctImgStatus = map[int32]string{ OctImgStatus = map[int32]string{
1: "未上传", 1: "未上传",
3: "制作完成", 3: "制作完成",
@@ -153,6 +158,10 @@ func GetModelNamesByType(t string) ([]string, error) {
} }


func GetDatasetsNames(ctx context.Context, collectorMap map[string]collector.AiCollector) ([]string, error) { func GetDatasetsNames(ctx context.Context, collectorMap map[string]collector.AiCollector) ([]string, error) {
return Datasets, nil
}

func GetDatasetsNamesSync(ctx context.Context, collectorMap map[string]collector.AiCollector) ([]string, error) {
var wg sync.WaitGroup var wg sync.WaitGroup
var errCh = make(chan interface{}, len(collectorMap)) var errCh = make(chan interface{}, len(collectorMap))
var errs []interface{} var errs []interface{}
@@ -225,6 +234,14 @@ func GetDatasetsNames(ctx context.Context, collectorMap map[string]collector.AiC
} }


func GetAlgorithms(ctx context.Context, collectorMap map[string]collector.AiCollector, resourceType string, taskType string, dataset string) ([]string, error) { func GetAlgorithms(ctx context.Context, collectorMap map[string]collector.AiCollector, resourceType string, taskType string, dataset string) ([]string, error) {
algorithm := AlgorithmsForDatasets[dataset]
if len(algorithm) != 0 {
return algorithm, nil
}
return nil, errors.New("not found")
}

func GetAlgorithmsSync(ctx context.Context, collectorMap map[string]collector.AiCollector, resourceType string, taskType string, dataset string) ([]string, error) {
var names []string var names []string
var wg sync.WaitGroup var wg sync.WaitGroup
var errCh = make(chan interface{}, len(collectorMap)) var errCh = make(chan interface{}, len(collectorMap))


Loading…
Cancel
Save