In [8]:

    

import numpy as np
import os

from plotly import __version__
import plotly.graph_objs as go
from plotly.offline import download_plotlyjs, init_notebook_mode, iplot
init_notebook_mode()


    

In [12]:

    

train_log_dir = '../complete/img_classification/convnet_train/'
traces = []
for f in os.listdir(train_log_dir):
    fsplit = f.split('_')
    method, lr, momentum = fsplit[1], fsplit[5], fsplit[6]
    log = np.load(os.path.join(train_log_dir, f))
    train_cost = log['train_cost']
    traces.append(go.Scatter(
        x = np.arange(len(train_cost)),
        y = train_cost,
        name = '_'.join([method, lr, momentum])))
data = go.Data(traces)
iplot(data)


    

In [ ]: