 Посмотреть на TensorFlow.org |  Запустить в Google Colab |  Посмотреть исходный код на GitHub |  Скачать блокнот |
Обзор
Этот учебник загружает CoreDNS метрики из Прометея сервера в tf.data.Dataset , а затем использует tf.keras для подготовки и вывода.
CoreDNS является DNS - сервер с акцентом на открытие услуг, а также широко используется в качестве части Kubernetes кластера. По этой причине он часто внимательно следит за операциями devops.
Это руководство является примером, который может быть использован разработчиками, стремящимися автоматизировать свои операции с помощью машинного обучения.
Настройка и использование
Установите необходимый пакет tensorflow-io и перезапустите среду выполнения.
import os
try:
%tensorflow_version 2.x
except Exception:
pass
TensorFlow 2.x selected.
pip install tensorflow-io
from datetime import datetime
import tensorflow as tf
import tensorflow_io as tfio
Установите и настройте CoreDNS и Prometheus
Для демонстрационных целей, сервер CoreDNS локально с портом 9053 открыт для получения DNS запросов и порт 9153 (defult) открытым , чтобы выставить метрики для соскабливания. Ниже приводится базовая конфигурация для CoreDNS Основной файл и доступен для скачивания :
.:9053 {
prometheus
whoami
}
Подробнее об установке можно найти на CoreDNS игровой документации .
curl -s -OL https://github.com/coredns/coredns/releases/download/v1.6.7/coredns_1.6.7_linux_amd64.tgztar -xzf coredns_1.6.7_linux_amd64.tgzcurl -s -OL https://raw.githubusercontent.com/tensorflow/io/master/docs/tutorials/prometheus/Corefilecat Corefile
.:9053 {
prometheus
whoami
}
# Run `./coredns` as a background process.
# IPython doesn't recognize `&` in inline bash cells.
get_ipython().system_raw('./coredns &')
Следующий шаг заключается в настройку Прометей сервер и использовать Прометей метрики скоблить CoreDNS, которые выставлены на порту 9153 сверху. prometheus.yml файл для настройки также доступны для загрузки :
curl -s -OL https://github.com/prometheus/prometheus/releases/download/v2.15.2/prometheus-2.15.2.linux-amd64.tar.gztar -xzf prometheus-2.15.2.linux-amd64.tar.gz --strip-components=1curl -s -OL https://raw.githubusercontent.com/tensorflow/io/master/docs/tutorials/prometheus/prometheus.ymlcat prometheus.yml
global:
scrape_interval: 1s
evaluation_interval: 1s
alerting:
alertmanagers:
- static_configs:
- targets:
rule_files:
scrape_configs:
- job_name: 'prometheus'
static_configs:
- targets: ['localhost:9090']
- job_name: "coredns"
static_configs:
- targets: ['localhost:9153']
# Run `./prometheus` as a background process.
# IPython doesn't recognize `&` in inline bash cells.
get_ipython().system_raw('./prometheus &')
Для того , чтобы показать некоторую активность, dig команда может быть использована для создания нескольких DNS - запросов к серверу CoreDNS , которая была установка:
sudo apt-get install -y -qq dnsutils
dig @127.0.0.1 -p 9053 demo1.example.org
; <<>> DiG 9.11.3-1ubuntu1.11-Ubuntu <<>> @127.0.0.1 -p 9053 demo1.example.org ; (1 server found) ;; global options: +cmd ;; Got answer: ;; ->>HEADER<<- opcode: QUERY, status: NOERROR, id: 53868 ;; flags: qr aa rd; QUERY: 1, ANSWER: 0, AUTHORITY: 0, ADDITIONAL: 3 ;; WARNING: recursion requested but not available ;; OPT PSEUDOSECTION: ; EDNS: version: 0, flags:; udp: 4096 ; COOKIE: 855234f1adcb7a28 (echoed) ;; QUESTION SECTION: ;demo1.example.org. IN A ;; ADDITIONAL SECTION: demo1.example.org. 0 IN A 127.0.0.1 _udp.demo1.example.org. 0 IN SRV 0 0 45361 . ;; Query time: 0 msec ;; SERVER: 127.0.0.1#9053(127.0.0.1) ;; WHEN: Tue Mar 03 22:35:20 UTC 2020 ;; MSG SIZE rcvd: 132
dig @127.0.0.1 -p 9053 demo2.example.org
; <<>> DiG 9.11.3-1ubuntu1.11-Ubuntu <<>> @127.0.0.1 -p 9053 demo2.example.org ; (1 server found) ;; global options: +cmd ;; Got answer: ;; ->>HEADER<<- opcode: QUERY, status: NOERROR, id: 53163 ;; flags: qr aa rd; QUERY: 1, ANSWER: 0, AUTHORITY: 0, ADDITIONAL: 3 ;; WARNING: recursion requested but not available ;; OPT PSEUDOSECTION: ; EDNS: version: 0, flags:; udp: 4096 ; COOKIE: f18b2ba23e13446d (echoed) ;; QUESTION SECTION: ;demo2.example.org. IN A ;; ADDITIONAL SECTION: demo2.example.org. 0 IN A 127.0.0.1 _udp.demo2.example.org. 0 IN SRV 0 0 42194 . ;; Query time: 0 msec ;; SERVER: 127.0.0.1#9053(127.0.0.1) ;; WHEN: Tue Mar 03 22:35:21 UTC 2020 ;; MSG SIZE rcvd: 132
Теперь это сервер CoreDNS, метрики которого собираются сервером Prometheus и готовы к использованию TensorFlow.
Создайте набор данных для метрик CoreDNS и используйте его в TensorFlow.
Создание набора данных для CoreDNS метрик , которые доступны с сервера PostgreSQL, может быть сделано с tfio.experimental.IODataset.from_prometheus . Нужны как минимум два аргумента. query передается на сервер Прометея , чтобы выбрать метрики и length это период вы хотите загрузить в Dataset.
Вы можете начать с "coredns_dns_request_count_total" и "5" (сек) , чтобы создать Dataset ниже. Так как ранее в учебнике два DNS - запросах были отправлены, то ожидается , что показатели для "coredns_dns_request_count_total" будут "2.0" в конце временного ряда:
dataset = tfio.experimental.IODataset.from_prometheus(
"coredns_dns_request_count_total", 5, endpoint="http://localhost:9090")
print("Dataset Spec:\n{}\n".format(dataset.element_spec))
print("CoreDNS Time Series:")
for (time, value) in dataset:
# time is milli second, convert to data time:
time = datetime.fromtimestamp(time // 1000)
print("{}: {}".format(time, value['coredns']['localhost:9153']['coredns_dns_request_count_total']))
Dataset Spec:
(TensorSpec(shape=(), dtype=tf.int64, name=None), {'coredns': {'localhost:9153': {'coredns_dns_request_count_total': TensorSpec(shape=(), dtype=tf.float64, name=None)} } })
CoreDNS Time Series:
2020-03-03 22:35:17: 2.0
2020-03-03 22:35:18: 2.0
2020-03-03 22:35:19: 2.0
2020-03-03 22:35:20: 2.0
2020-03-03 22:35:21: 2.0
Дальнейшее изучение спецификации набора данных:
(
TensorSpec(shape=(), dtype=tf.int64, name=None),
{
'coredns': {
'localhost:9153': {
'coredns_dns_request_count_total': TensorSpec(shape=(), dtype=tf.float64, name=None)
}
}
}
)
Очевидно , что набор данных состоит из (time, values) кортежа , где values поля является Python ДИКТ расширен в:
"job_name": {
"instance_name": {
"metric_name": value,
},
}
В приведенном выше примере, 'coredns' это имя задания, 'localhost:9153' это имя экземпляра, и 'coredns_dns_request_count_total' метрический имя. Обратите внимание, что в зависимости от используемого запроса Prometheus может быть возвращено несколько заданий/экземпляров/метрик. Это также причина, по которой python dict используется в структуре набора данных.
Возьмем другой запрос "go_memstats_gc_sys_bytes" в качестве примера. Так как CoreDNS и Прометей написаны в Golang, "go_memstats_gc_sys_bytes" Метрика доступна как для "coredns" работы и "prometheus" работы:
dataset = tfio.experimental.IODataset.from_prometheus(
"go_memstats_gc_sys_bytes", 5, endpoint="http://localhost:9090")
print("Time Series CoreDNS/Prometheus Comparision:")
for (time, value) in dataset:
# time is milli second, convert to data time:
time = datetime.fromtimestamp(time // 1000)
print("{}: {}/{}".format(
time,
value['coredns']['localhost:9153']['go_memstats_gc_sys_bytes'],
value['prometheus']['localhost:9090']['go_memstats_gc_sys_bytes']))
Time Series CoreDNS/Prometheus Comparision: 2020-03-03 22:35:17: 2385920.0/2775040.0 2020-03-03 22:35:18: 2385920.0/2775040.0 2020-03-03 22:35:19: 2385920.0/2775040.0 2020-03-03 22:35:20: 2385920.0/2775040.0 2020-03-03 22:35:21: 2385920.0/2775040.0
Созданный Dataset готов быть передано tf.keras непосредственно либо для обучения или вывода целей в настоящее время.
Используйте набор данных для обучения модели
С метриками Dataset созданы, можно непосредственно передать Dataset в tf.keras для обучения модели или умозаключения.
В демонстрационных целях в этом руководстве будет использоваться очень простая модель LSTM с 1 функцией и 2 шагами в качестве входных данных:
n_steps, n_features = 2, 1
simple_lstm_model = tf.keras.models.Sequential([
tf.keras.layers.LSTM(8, input_shape=(n_steps, n_features)),
tf.keras.layers.Dense(1)
])
simple_lstm_model.compile(optimizer='adam', loss='mae')
Используемый набор данных — это значение «go_memstats_sys_bytes» для CoreDNS с 10 выборками. Однако, так как скользящее окно window=n_steps и shift=1 образуется, необходимы дополнительные образцы (для любых двух элементов consecute, первым берутся в качестве x , а второй берутся как y для обучения). Всего 10 + n_steps - 1 + 1 = 12 секунд.
Значение данных также масштабировать до [0, 1] .
n_samples = 10
dataset = tfio.experimental.IODataset.from_prometheus(
"go_memstats_sys_bytes", n_samples + n_steps - 1 + 1, endpoint="http://localhost:9090")
# take go_memstats_gc_sys_bytes from coredns job
dataset = dataset.map(lambda _, v: v['coredns']['localhost:9153']['go_memstats_sys_bytes'])
# find the max value and scale the value to [0, 1]
v_max = dataset.reduce(tf.constant(0.0, tf.float64), tf.math.maximum)
dataset = dataset.map(lambda v: (v / v_max))
# expand the dimension by 1 to fit n_features=1
dataset = dataset.map(lambda v: tf.expand_dims(v, -1))
# take a sliding window
dataset = dataset.window(n_steps, shift=1, drop_remainder=True)
dataset = dataset.flat_map(lambda d: d.batch(n_steps))
# the first value is x and the next value is y, only take 10 samples
x = dataset.take(n_samples)
y = dataset.skip(1).take(n_samples)
dataset = tf.data.Dataset.zip((x, y))
# pass the final dataset to model.fit for training
simple_lstm_model.fit(dataset.batch(1).repeat(10), epochs=5, steps_per_epoch=10)
Train for 10 steps Epoch 1/5 10/10 [==============================] - 2s 150ms/step - loss: 0.8484 Epoch 2/5 10/10 [==============================] - 0s 10ms/step - loss: 0.7808 Epoch 3/5 10/10 [==============================] - 0s 10ms/step - loss: 0.7102 Epoch 4/5 10/10 [==============================] - 0s 11ms/step - loss: 0.6359 Epoch 5/5 10/10 [==============================] - 0s 11ms/step - loss: 0.5572 <tensorflow.python.keras.callbacks.History at 0x7f1758f3da90>
Приведенная выше обученная модель в действительности не очень полезна, так как сервер CoreDNS, который был настроен в этом руководстве, не имеет никакой рабочей нагрузки. Однако это рабочий конвейер, который можно использовать для загрузки метрик с настоящих рабочих серверов. Затем модель можно было бы улучшить, чтобы решить реальную проблему автоматизации devops.