Function LoadVanillaModel

model/model.go:177–277

(
	errorHandler *cberrors.ErrorsContainer,
	logger *cblog.Logger,
	dir string,
	loss Loss,
	optimizer optimizer.Optimizer,
	sessionOptions ...*for_core_protos_go_proto.ConfigProto,
)

}

func LoadVanillaModel(
	errorHandler *cberrors.ErrorsContainer,
	logger *cblog.Logger,
	dir string,
	loss Loss,
	optimizer optimizer.Optimizer,
	sessionOptions ...*for_core_protos_go_proto.ConfigProto,
) (*TfkgModel, error) {
	logger.InfoF("model", "Loading vanilla model. If anything goes wrong python error messages will be printed out.")

	tempDir := filepath.Join(os.TempDir(), "/tfkg")

	e := os.MkdirAll(tempDir, os.ModePerm)
	if e != nil {
		errorHandler.Error(e)
		return nil, e
	}

	config := vanillaPythonConfig{
		ModelDir:  dir,
		SaveDir:   filepath.Join(tempDir, tempModelDir),
		Loss:      string(loss),
		Optimizer: optimizer.GetKerasLayerConfig(),
	}

	configBytes, e := json.Marshal(config)
	if e != nil {
		errorHandler.Error(e)
		return nil, e
	}

	tempPythonPath := filepath.Join(tempDir, "tfkg_create_model.py")

	e = ioutil.WriteFile(tempPythonPath, []byte(GetVanillaPythonCode()), os.ModePerm)
	if e != nil {
		errorHandler.Error(e)
		return nil, e
	}
	defer os.Remove(tempPythonPath)

	tempConfigPath := filepath.Join(tempDir, "tfkg_config.json")

	e = ioutil.WriteFile(tempConfigPath, configBytes, os.ModePerm)
	if e != nil {
		errorHandler.Error(e)
		return nil, e
	}
	defer os.Remove(tempPythonPath)

	cmd := exec.Command("python", tempPythonPath, tempConfigPath)
	output, e := cmd.CombinedOutput()
	if e != nil {
		errorHandler.Error(e)
	}
	fmt.Println(string(output))
	if e != nil {
		return nil, e
	}