neptune integration update

Author: stared

examples/neptune.py

@@ -1,5 +1,5 @@
 # TO START:
-# pip install neptune-client, livelossplot
+# pip install neptune, livelossplot
 # export environment variables
 # enjoy results
 
@@ -13,22 +13,26 @@
 
 
 def main():
-    api_token = os.environ.get('NEPTUNE_API_TOKEN')
-    project_qualified_name = os.environ.get('NEPTUNE_PROJECT_NAME')
-    logger = NeptuneLogger(api_token=api_token, project_qualified_name=project_qualified_name)
+    api_token = os.environ.get("NEPTUNE_API_TOKEN")
+    project_qualified_name = os.environ.get("NEPTUNE_PROJECT")
+    logger = NeptuneLogger(
+        api_token=api_token,
+        project_qualified_name=project_qualified_name,
+        tags=["example", "livelossplot"],
+    )
     liveplot = PlotLosses(outputs=[logger])
     for i in range(20):
         liveplot.update(
             {
-                'accuracy': 1 - np.random.rand() / (i + 2.),
-                'val_accuracy': 1 - np.random.rand() / (i + 0.5),
-                'mse': 1. / (i + 2.),
-                'val_mse': 1. / (i + 0.5)
+                "accuracy": 1 - np.random.rand() / (i + 2.0),
+                "val_accuracy": 1 - np.random.rand() / (i + 0.5),
+                "mse": 1.0 / (i + 2.0),
+                "val_mse": 1.0 / (i + 0.5),
             }
         )
         liveplot.send()
-        sleep(.5)
+        sleep(0.5)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

livelossplot/outputs/neptune_logger.py

@@ -7,24 +7,33 @@ class NeptuneLogger(BaseOutput):
     """See: https://github.com/neptune-ai/neptune-client
     YOUR_API_TOKEN and USERNAME/PROJECT_NAME
     """
-    def __init__(self, api_token: Optional[str] = None, project_qualified_name: Optional[str] = None, **kwargs):
+
+    def __init__(
+        self,
+        api_token: Optional[str] = None,
+        project_qualified_name: Optional[str] = None,
+        **kwargs
+    ):
         """Set secrets and create experiment
         Args:
             api_token: your api token, you can create NEPTUNE_API_TOKEN environment variable instead
             project_qualified_name: <user>/<project>, you can create NEPTUNE_PROJECT environment variable instead
             **kwargs: keyword args, that will be passed to create_experiment function
         """
         import neptune
+
         self.neptune = neptune
-        self.neptune.init(api_token=api_token, project_qualified_name=project_qualified_name)
-        self.experiment = self.neptune.create_experiment(**kwargs)
+        self.run = self.neptune.init_run(
+            api_token=api_token, project=project_qualified_name, **kwargs
+        )
 
     def close(self):
         """Close connection"""
-        self.neptune.stop()
+        if hasattr(self, "run"):
+            self.run.stop()
 
     def send(self, logger: MainLogger):
         """Send metrics collected in last step to neptune server"""
         for name, log_items in logger.log_history.items():
             last_log_item = log_items[-1]
-            self.neptune.send_metric(name, x=last_log_item.step, y=last_log_item.value)
+            self.run[name].append(value=last_log_item.value, step=last_log_item.step)