From 95eb4705afc5f8fe8c32f887ee19e1ec44cf7586 Mon Sep 17 00:00:00 2001 From: Etienne Perot Date: Thu, 18 Jan 2024 13:31:03 -0800 Subject: [PATCH] Remove `GradientDecisionTree` Tensorflow benchmark. It often times out for me. PiperOrigin-RevId: 599608747 --- test/benchmarks/ml/tensorflow_test.go | 1 - 1 file changed, 1 deletion(-) diff --git a/test/benchmarks/ml/tensorflow_test.go b/test/benchmarks/ml/tensorflow_test.go index 58d4f7fd2..d95a3591d 100644 --- a/test/benchmarks/ml/tensorflow_test.go +++ b/test/benchmarks/ml/tensorflow_test.go @@ -36,7 +36,6 @@ func BenchmarkTensorflowDashboard(b *testing.B) { // See: https://github.com/aymericdamien/TensorFlow-Examples func BenchmarkTensorflow(b *testing.B) { workloads := map[string]string{ - "GradientDecisionTree": "2_BasicModels/gradient_boosted_decision_tree.py", "Kmeans": "2_BasicModels/kmeans.py", "LogisticRegression": "2_BasicModels/logistic_regression.py", "NearestNeighbor": "2_BasicModels/nearest_neighbor.py",