template

Documentation

Index

• Constants
• func BuildHeteroPredictingConf(param HeteroPredictingParam) (string, string, error)
• func BuildHeteroTrainingConf(param HeteroTrainingParam) (string, string, error)
• func BuildHomoPredictingConf(param HomoPredictingParam) (string, string, error)
• func BuildHomoTrainingConf(param HomoTrainingParam) (string, string, error)
• func BuildPsiConf(param PSIParam) (string, string, error)
• func BuildTrainingConfGeneralStr(param GeneralTrainingParam) (string, error)
• type GeneralTrainingParam
• type HeteroAlgorithmType
• type HeteroPredictingParam
• type HeteroTrainingParam
• type HomoAlgorithmType
• type HomoPredictingParam
• type HomoTrainingParam
• type PSIParam
• type PartyDataInfo

Constants

const PSIConf = `` /* 1190-byte string literal not displayed */

const PSIDSL = `` /* 933-byte string literal not displayed */

const PSIHostParamTemplate = `
{
  "reader_0": {
    "table": {
      "name": "%s",
      "namespace": "%s"
    }
  }
}
`

Functions

func BuildHeteroPredictingConf

func BuildHeteroPredictingConf(param HeteroPredictingParam) (string, string, error)

BuildHeteroPredictingConf returns the FATE job conf and dsl from the specified param

func BuildHeteroTrainingConf

func BuildHeteroTrainingConf(param HeteroTrainingParam) (string, string, error)

BuildHeteroTrainingConf returns the FATE job conf and dsl from the specified param

func BuildHomoPredictingConf

func BuildHomoPredictingConf(param HomoPredictingParam) (string, string, error)

BuildHomoPredictingConf returns the FATE job conf and dsl from the specified param

func BuildHomoTrainingConf

func BuildHomoTrainingConf(param HomoTrainingParam) (string, string, error)

BuildHomoTrainingConf returns the FATE job conf and dsl from the specified param

func BuildPsiConf

func BuildPsiConf(param PSIParam) (string, string, error)

BuildPsiConf returns the DSL and Conf files

func BuildTrainingConfGeneralStr

func BuildTrainingConfGeneralStr(param GeneralTrainingParam) (string, error)

BuildTrainingConfGeneralStr returns a json string, which is a part of the final FATE job conf file. Including "dsl_version", "initiator", "role" and "job_parameters".

Types

type GeneralTrainingParam

type GeneralTrainingParam struct {
	Guest PartyDataInfo
	Hosts []PartyDataInfo
}

GeneralTrainingParam contains common parameters for creating all kins of jobs

type HeteroAlgorithmType

type HeteroAlgorithmType uint8

HeteroAlgorithmType is the enum of vertical algorithm types

const (
	HeteroAlgorithmTypeUnknown HeteroAlgorithmType = iota
	HeteroAlgorithmTypeLR
	HeteroAlgorithmTypeSBT
)

type HeteroPredictingParam

type HeteroPredictingParam struct {
	Guest        PartyDataInfo
	Hosts        []PartyDataInfo
	ModelID      string
	ModelVersion string
}

HeteroPredictingParam contains parameters for creating a predicting job for a vertical model

type HeteroTrainingParam

type HeteroTrainingParam struct {
	Guest             PartyDataInfo
	Hosts             []PartyDataInfo
	LabelName         string
	ValidationEnabled bool
	ValidationPercent uint
	Type              HeteroAlgorithmType
}

HeteroTrainingParam contains parameters for a vertical job

type HomoAlgorithmType

type HomoAlgorithmType uint8

HomoAlgorithmType is the enum of horizontal algorithm types

const (
	HomoAlgorithmTypeUnknown HomoAlgorithmType = iota
	HomoAlgorithmTypeLR
	HomoAlgorithmTypeSBT
)

type HomoPredictingParam

type HomoPredictingParam struct {
	Role          string
	ModelID       string
	ModelVersion  string
	PartyDataInfo PartyDataInfo
}

HomoPredictingParam contains parameters for creating a predicting job for a horizontal model

type HomoTrainingParam

type HomoTrainingParam struct {
	Guest             PartyDataInfo
	Hosts             []PartyDataInfo
	LabelName         string
	ValidationEnabled bool
	ValidationPercent uint
	Type              HomoAlgorithmType
}

HomoTrainingParam contains parameters for a horizontal job

type PSIParam

type PSIParam struct {
	Guest PartyDataInfo
	Hosts []PartyDataInfo
}

PSIParam contains parameters for a PSI job

type PartyDataInfo

type PartyDataInfo struct {
	PartyID        string
	TableName      string
	TableNamespace string
}

PartyDataInfo is contains party info and its data info