goscore

type ActivationFunction func(float64) float64

type Contribution struct {
	From   string  `xml:"from,attr"`
	Weight float64 `xml:"weight,attr"`
}

type DerivedField struct {
	DataType     string `xml:"dataType,attr"`
	FieldRef     FieldRef
	NormDiscrete NormDiscrete
}

type FieldRef struct {
	Field string `xml:"field,attr"`
}

type GradientBoostedModel struct {
	Version  string  `xml:"version,attr"`
	Trees    []Node  `xml:"MiningModel>Segmentation>Segment>MiningModel>Segmentation>Segment>TreeModel"`
	Target   target  `xml:"MiningModel>Segmentation>Segment>MiningModel>Targets>Target"`
	Constant float64 `xml:"MiningModel>Segmentation>Segment>MiningModel>Output>OutputField>Apply>Constant"`
}

type LogisticRegression struct {
	// struct xml:PMML>RegressionModel
	NormalizationMethod string            `xml:"normalizationMethod,attr"`
	Fields              []MiningField     `xml:"MiningSchema>MiningField"`
	RegressionTable     []RegressionTable `xml:"RegressionTable"`
}

type MiningField struct {
	Name string `xml:"name,attr"`
}

type NeuralInputs struct {
	Input []Neuron `xml:"NeuralInput"`
}

type NeuralLayer struct {
	Neuron              []Neuron `xml:"Neuron"`
	ActivationFunction  string   `xml:"activationFunction,attr"`
	NormalizationMethod string   `xml:"normalizationMethod,attr"`
	Threshold           float64  `xml:"threshold,attr"`
}

type NeuralNetWorkStructure struct {
}

type NeuralNetwork struct {
	XMLName xml.Name
	//Struct  NeuralNetWorkStructure `xml:"PMML>NeuralNetwork"`
	InputLayer          NeuralInputs  `xml:"NeuralInputs"`
	NeuralOutputs       NeuralOutputs `xml:"NeuralOutputs"`
	OutputLayer         Outputs       `xml:"Output"`
	Fields              []MiningField `xml:"MiningSchema>MiningField"`
	Layers              []NeuralLayer `xml:"NeuralLayer"`
	ActivationFunction  string        `xml:"activationFunction,attr"`
	NormalizationMethod string        `xml:"normalizationMethod,attr"`
	Threshold           float64       `xml:"threshold,attr"`
}

type NeuralOutput struct {
	OutputNeuron string       `xml:"outputNeuron,attr"`
	DerivedField DerivedField `xml:"DerivedField"`
}

type NeuralOutputs struct {
	NeuralOutput []NeuralOutput `xml:"NeuralOutput"`
}

type Neuron struct {
	Id               string         `xml:"id,attr"`
	DerivedFieldType DerivedField   `xml:"DerivedField"`
	Bias             float64        `xml:"bias,attr"`
	Contribution     []Contribution `xml:"Con"`
}

type Node struct {
	XMLName            xml.Name
	Attrs              []xml.Attr         `xml:",any,attr"`
	Content            []byte             `xml:",innerxml"`
	Nodes              []Node             `xml:",any"`
	True               truePredicate      `xml:"True"`
	DummyMiningSchema  dummyMiningSchema  `xml:"MiningSchema"`
	SimplePredicate    SimplePredicate    `xml:"SimplePredicate"`
	SimpleSetPredicate SimpleSetPredicate `xml:"SimpleSetPredicate"`
}

type NormDiscrete struct {
	Value string `xml:"value,attr"`
}

type NormalizationMethod func(...float64) []float64

type NormalizationMethodMap func(map[string]float64) (map[string]float64, error)

type NumericPredictor struct {
	// struct xml:PMML>RegressionModel>RegressionTable>NumbericPredictor
	Name        string  `xml:"name,attr"`
	Coefficient float64 `xml:"coefficient,attr"`
}

type OutputField struct {
	Name  string `xml:"name,attr"`
	Value string `xml:"value,attr"`
}

type Outputs struct {
	OutputField []OutputField `xml:"OutputField"`
}

type PMMLLR struct {
	// struct xml:PMML
	LogisticRegression LogisticRegression `xml:"RegressionModel"`
}

type PMMLNN struct {
	NeuralNetwork NeuralNetwork `xml:"NeuralNetwork"`
}

type RandomForest struct {
	XMLName xml.Name
	Trees   []Node `xml:"MiningModel>Segmentation>Segment>TreeModel"`
}

type RegressionTable struct {
	// struct xml:PMML>RegressionModel>RegressionTable
	Intercept           float64            `xml:"intercept,attr"`
	TargetCategory      string             `xml:"targetCategory,attr"`
	NumericPredictor    []NumericPredictor `xml:"NumericPredictor"`
	NumericPredictorMap *map[string]float64
}

type SimplePredicate struct {
	Field    string `xml:"field,attr"`
	Operator string `xml:"operator,attr"`
	Value    string `xml:"value,attr"`
}

type SimpleSetPredicate struct {
	Field    string `xml:"field,attr"`
	Operator string `xml:"booleanOperator,attr"`
	Values   string `xml:"Array"`
}