30 lines
997 B
ObjectPascal
30 lines
997 B
ObjectPascal
// В этом примере Surface показывает,
|
||
// как модель делит плоскость на области классов.
|
||
|
||
uses MLABC, PlotML;
|
||
|
||
begin
|
||
var (X, y) := Datasets.MakeMoons(
|
||
n := 300,
|
||
noise := 0.18,
|
||
seed := 42
|
||
);
|
||
|
||
var model := new DecisionTreeClassifier(maxDepth := 5, minSamplesSplit := 6, minSamplesLeaf := 3, seed := 42);
|
||
model.Fit(X, y);
|
||
var acc := ClassificationMetrics.Accuracy(y, model.Predict(X));
|
||
|
||
var x1 := X.Col(0);
|
||
var x2 := X.Col(1);
|
||
var labels := y;
|
||
|
||
Println('Граница решений для DecisionTreeClassifier');
|
||
Println;
|
||
Println('Цветной фон показывает области, которые модель относит к разным классам.');
|
||
Println($'Accuracy = {acc:F3}');
|
||
|
||
Plot.Surface(x1, x2, 80, 80, G -> model.Predict(G), Palettes.Pastel);
|
||
Plot.Points(x1, x2, labels, size := 6);
|
||
Plot.Title := $'DecisionTreeClassifier: decision boundary (Acc = {acc:F3})';
|
||
end.
|