使用深度学习组件,需要用到TensorFlow环境,这个环境不是默认安装的。需要借助Alink插件下载器(PluginDownloader),使用介绍参见 https://www.yuque.com/pinshu/alink_guide/plugin_downloader 。
下面做几个实验,深入了解一下。先运行如下代码,获取默认的下载文件夹:
System.out.println(AlinkGlobalConfiguration.getPluginDir());
运行结果如下,
plugins
即,默认的下载文件夹是当前运行路径的子文件夹,名称为“plugins”。用户可以使用AlinkGlobalConfiguration.setPluginDir方法,设置新的下载文件夹。
再运行如下代码,获取是否自动进行插件下载的属性。
System.out.print("Auto Plugin Download : "); System.out.println(AlinkGlobalConfiguration.getAutoPluginDownload());
运行结果如下,
Auto Plugin Download : true
即,自动下载属性的默认值为“true”。用户可以使用AlinkGlobalConfiguration.setAutoPluginDownload方法更改设置。
了解Alink关于插件下载器的基本设置,在使用过程中可以使用两种方式:
我们看一下当前插件列表里有哪些内容,哪些和深度学习相关,代码如下。
PluginDownloader downloader = AlinkGlobalConfiguration.getPluginDownloader(); System.out.println(downloader.listAvailablePlugins());
运行结果如下,
[derby, hadoop, hive, kafka, mysql, odps, oss, s3-hadoop, s3-presto, sqlite, tf_predictor_linux, tf_predictor_macosx, tf_predictor_windows, base_cased_ckpt, base_cased_saved_model, base_cased_vocab, base_chinese_ckpt, base_chinese_saved_model, base_chinese_vocab, base_multilingual_cased_ckpt, base_multilingual_cased_saved_model, base_multilingual_cased_vocab, base_uncased_ckpt, base_uncased_saved_model, base_uncased_vocab, tf115_python_env_linux, tf115_python_env_macosx, tf115_python_env_windows, tf231_python_env_linux, tf231_python_env_macosx, tf231_python_env_windows]
其中“tf231_python_env_macosx”为TensorFlow 2.3.1 适合 Mac OS的版本;“base_chinese_ckpt”为BERT组件所需的中文预训练模型文件。