Ejemplo n.º 1
0
    def _dump_metadata(self):
        display.display(display.Link(
            'https://console.cloud.google.com/mlengine/models/{}/versions/{}?project={}'.format(
                self._model_id, self._version_id, self._project_id),
            'Version Details'
        ))
        display.display(display.Markdown('''
## Online Prediction

### REST endpoint
The REST endpoint for online prediction is as follows:
```
POST https://ml.googleapis.com/v1/{}:predict
```
Try the REST endpoint in [Google OAuth 2.0 Playgound](https://developers.google.com/oauthplayground/#step3\
&apisSelect=https://www.googleapis.com/auth/cloud-platform&postData={{"instances":[]}}\
&url=https://ml.googleapis.com/v1/{}:predict&content_type=application/json&http_method=POST).

### GCloud command
```bash
gcloud ml-engine predict --model {}  \
                   --version {} \
                   --json-instances instances.json
```
        '''.format(self._version_name, self._version_name, self._model_id, self._version_id)))
Ejemplo n.º 2
0
    def test_display_markdown(self, mock_open, mock_os, mock_json):
        mock_os.path.isfile.return_value = False

        display.display(display.Markdown('# test'))

        mock_json.dump.assert_called_with(
            {
                'outputs': [{
                    'type': 'markdown',
                    'source': '# test',
                    'storage': 'inline'
                }]
            }, mock.ANY)
Ejemplo n.º 3
0
    def test_display_markdown_append(self, mock_open, mock_os, mock_json):
        mock_os.path.isfile.return_value = True
        mock_json.load.return_value = {
            'outputs': [{
                'type': 'markdown',
                'source': '# test 1',
                'storage': 'inline'
            }]
        }

        display.display(display.Markdown('# test 2'))

        mock_json.dump.assert_called_with(
            {
                'outputs': [{
                    'type': 'markdown',
                    'source': '# test 1',
                    'storage': 'inline'
                }, {
                    'type': 'markdown',
                    'source': '# test 2',
                    'storage': 'inline'
                }]
            }, mock.ANY)
Ejemplo n.º 4
0
 def test___repr__(self, mock_open, mock_os, mock_json):
     self.assertEqual('# Title', str(display.Markdown('# Title')))
     self.assertEqual('Open Tensorboard at: gs://trained/model/',
                      str(display.Tensorboard('gs://trained/model/')))
     self.assertEqual('title: https://test/uri',
                      str(display.Link('https://test/uri', 'title')))