|
29 | 29 | "import sys\n", |
30 | 30 | "import pickle\n", |
31 | 31 | "import gzip\n", |
32 | | - "from typing import List, Dict, Callable\n", |
33 | | - "from collections import defaultdict, Counter\n", |
| 32 | + "from typing import Dict\n", |
34 | 33 | "\n", |
35 | 34 | "import h5py\n", |
36 | 35 | "import numpy as np\n", |
37 | 36 | "\n", |
38 | 37 | "\n", |
39 | | - "from ml4h.defines import StorageType\n", |
40 | 38 | "from ml4h.arguments import parse_args\n", |
41 | 39 | "from ml4h.TensorMap import TensorMap, Interpretation\n", |
42 | 40 | "from ml4h.tensor_generators import test_train_valid_tensor_generators\n", |
43 | | - "from ml4h.models.train import train_model_from_generators\n", |
44 | | - "from ml4h.models.model_factory import make_multimodal_multitask_model\n", |
45 | | - "from ml4h.models.inspect import plot_and_time_model\n", |
46 | 41 | "from ml4h.recipes import compare_multimodal_scalar_task_models, train_multimodal_multitask\n", |
47 | 42 | "\n", |
48 | | - "%matplotlib inline\n", |
49 | | - "import matplotlib.pyplot as plt\n", |
50 | | - "from matplotlib import gridspec" |
| 43 | + "%matplotlib inline" |
51 | 44 | ] |
52 | 45 | }, |
53 | 46 | { |
|
203 | 196 | "sys.argv = ['train', \n", |
204 | 197 | " '--tensors', HD5_FOLDER, \n", |
205 | 198 | " '--input_tensors', 'mnist.mnist_image',\n", |
206 | | - " '--output_tensors', 'mnit.mnist_label',\n", |
| 199 | + " '--output_tensors', 'mnist.mnist_label',\n", |
207 | 200 | " '--batch_size', '16',\n", |
208 | 201 | " '--epochs', '12',\n", |
209 | 202 | " '--output_folder', OUTPUT_FOLDER,\n", |
|
215 | 208 | "sys.argv = ['train',\n", |
216 | 209 | " '--tensors', HD5_FOLDER,\n", |
217 | 210 | " '--input_tensors', 'mnist.mnist_image',\n", |
218 | | - " '--output_tensors', 'mnit.mnist_label',\n", |
| 211 | + " '--output_tensors', 'mnist.mnist_label',\n", |
219 | 212 | " '--activation', 'mish',\n", |
220 | 213 | " '--dense_blocks', '64', '64', '64',\n", |
221 | 214 | " '--batch_size', '16',\n", |
|
243 | 236 | "sys.argv = ['compare_scalar', \n", |
244 | 237 | " '--tensors', HD5_FOLDER, \n", |
245 | 238 | " '--input_tensors', 'mnist.mnist_image',\n", |
246 | | - " '--output_tensors', 'mnit.mnist_label',\n", |
| 239 | + " '--output_tensors', 'mnist.mnist_label',\n", |
247 | 240 | " '--id', 'mnist_model_comparison',\n", |
248 | 241 | " '--output_folder', OUTPUT_FOLDER,\n", |
249 | 242 | " '--model_files', f'{OUTPUT_FOLDER}learn_2d_cnn/learn_2d_cnn.h5',\n", |
|
0 commit comments