Perform regression with all points

1 files changed, 4 insertions, 6 deletions

diff --git a/research/lesci/plot.go b/research/lesci/plot.go
index f649833..fba6939 100644
--- a/research/lesci/plot.go
+++ b/research/lesci/plot.go
@@ -41,12 +41,6 @@ func (e *Experiment) Plot(db *sql.DB) error {
 			return err
 		}
 
-		if mean < -20 || mean > 0 {
-			continue
-		}
-
-		pts = append(pts, plotter.XY{X: tok, Y: mean})
-
 		if treat {
 			inX = append(inX, tok)
 			inY = append(inY, mean)
@@ -54,6 +48,10 @@ func (e *Experiment) Plot(db *sql.DB) error {
 			oovX = append(oovX, tok)
 			oovY = append(oovY, mean)
 		}
+
+		if mean >= -20 && mean <= 0 {
+			pts = append(pts, plotter.XY{X: tok, Y: mean})
+		}
 	}
 
 	if err := rows.Err(); err != nil {