Add pydot package (#956) · JasonYangCode/docker-python@c4bd2a9 · GitHub

JasonYangCode / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 1

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit c4bd2a9

Browse filesBrowse the repository at this point in the historyBrowse files

rosbo

authored

Add pydot package (Kaggle#956)

This is used by keras plot_model() function. Included a test to prevent regression. Improved the existing tf test with extra assertions. The issue was raised by a user here: https://www.kaggle.com/c/jane-street-market-prediction/discussion/214494#1184233

1 parent ddc0c00 commit c4bd2a9
Copy full SHA for c4bd2a9

2 files changed

+11-3Lines changed: 11 additions & 3 deletions

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• Dockerfile
• tests
  • test_tensorflow.py

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎Dockerfile‎

Copy file name to clipboardExpand all lines: Dockerfile

+1Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
@@ -80,6 +80,7 @@ RUN apt-get install -y libfreetype6-dev && \
80	80	pip install xgboost && \
81	81	# Pinned to match GPU version. Update version together.
82	82	pip install lightgbm==3.1.1 && \
	83	+ pip install pydot && \
83	84	pip install keras && \
84	85	pip install keras-tuner && \
85	86	pip install flake8 && \

Collapse file

‎tests/test_tensorflow.py‎

Copy file name to clipboardExpand all lines: tests/test_tensorflow.py

+10-3Lines changed: 10 additions & 3 deletions

Original file line number	Diff line number	Diff line change
@@ -1,4 +1,5 @@
1	1	import unittest
	2	+ import os.path
2	3
3	4	import numpy as np
4	5	import tensorflow as tf
@@ -37,8 +38,13 @@ def test_tf_keras(self):
37	38	metrics=['accuracy'])
38	39
39	40	model.fit(x_train, y_train, epochs=1)
40		- model.evaluate(x_test, y_test)
41		-
	41	+
	42	+ result = model.evaluate(x_test, y_test)
	43	+ self.assertEqual(2, len(result))
	44	+
	45	+ # exercices pydot path.
	46	+ tf.keras.utils.plot_model(model, to_file="tf_plot_model.png")
	47	+ self.assertTrue(os.path.isfile("tf_plot_model.png"))
42	48
43	49	def test_lstm(self):
44	50	x_train = np.random.random((100, 28, 28))
@@ -58,7 +64,8 @@ def test_lstm(self):
58	64	metrics=['accuracy'])
59	65
60	66	model.fit(x_train, y_train, epochs=1)
61		- model.evaluate(x_test, y_test)
	67	+ result = model.evaluate(x_test, y_test)
	68	+ self.assertEqual(2, len(result))
62	69
63	70	@gpu_test
64	71	def test_gpu(self):

0 commit comments

Comments

0 (0)