package/kmeans
1
0
Fork 0
mirror of https://github.com/muesli/kmeans.git synced 2025-09-27 03:56:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Never panic in Plot, return an error instead

Browse Source
This commit is contained in:
Christian Muehlhaeuser
2018-06-06 00:49:26 +02:00
parent 8f4a4acbca
commit 144a85a37f
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
plotter.go
View File

@@ -13,7 +13,7 @@ import (
// The Plotter interface lets you implement your own plotters // The Plotter interface lets you implement your own plotters
type Plotter interface { type Plotter interface {
Plot(cc clusters.Clusters, iteration int) Plot(cc clusters.Clusters, iteration int) error
} }
// SimplePlotter is the default standard plotter for 2-dimensional data sets // SimplePlotter is the default standard plotter for 2-dimensional data sets
@@ -34,8 +34,8 @@ var colors = []drawing.Color{
drawing.ColorFromHex("4b7509"), drawing.ColorFromHex("4b7509"),
} }
// Plot draw a 2-dimensional data set into a PNG file named {iteration}.png // Plot draw a 2-dimensional data set into a PNG file named {k_iteration}.png
func (p SimplePlotter) Plot(cc clusters.Clusters, iteration int) { func (p SimplePlotter) Plot(cc clusters.Clusters, iteration int) error {
var series []chart.Series var series []chart.Series
// draw data points // draw data points
@@ -83,10 +83,7 @@ func (p SimplePlotter) Plot(cc clusters.Clusters, iteration int) {
buffer := bytes.NewBuffer([]byte{}) buffer := bytes.NewBuffer([]byte{})
err := graph.Render(chart.PNG, buffer) err := graph.Render(chart.PNG, buffer)
if err != nil { if err != nil {
panic(err) return err
}
err = ioutil.WriteFile(fmt.Sprintf("%d_%d.png", len(cc), iteration), buffer.Bytes(), 0644)
if err != nil {
panic(err)
} }
return ioutil.WriteFile(fmt.Sprintf("%d_%d.png", len(cc), iteration), buffer.Bytes(), 0644)
} }