remove config nn directories

config/APAN.yml

-sampling:
-  - layer: 1
-    neighbor: 
-      - 10
-    strategy: 'recent'
-    prop_time: False
-    history: 1
-    duration: 0
-    num_thread: 32
-    no_neg: True
-memory: 
-  - type: 'node'
-    dim_time: 100
-    deliver_to: 'neighbors'
-    mail_combine: 'last'
-    memory_update: 'transformer'
-    attention_head: 2
-    mailbox_size: 10
-    combine_node_feature: False
-    dim_out: 100
-gnn:
-  - arch: 'identity'
-train:
-  - epoch: 100
-    batch_size: 600
-    lr: 0.0001
-    dropout: 0.1
-    att_dropout: 0.1
-    # all_on_gpu: True
\ No newline at end of file

config/DySAT.yml

-sampling:
-  - layer: 2
-    neighbor: 
-      - 10
-      - 10
-    strategy: 'uniform'
-    prop_time: True
-    history: 3
-    duration: 10000
-    num_thread: 32
-memory: 
-  - type: 'none'
-    dim_out: 0
-gnn:
-  - arch: 'transformer_attention'
-    layer: 2
-    att_head: 2
-    dim_time: 0
-    dim_out: 100
-    combine: 'rnn'
-train:
-  - epoch: 50
-    batch_size: 600
-    lr: 0.0001
-    dropout: 0.1
-    att_dropout: 0.1
-    all_on_gpu: True
\ No newline at end of file

config/JODIE.yml

-sampling:
-  - no_sample: True
-    history: 1
-memory: 
-  - type: 'node'
-    dim_time: 100
-    deliver_to: 'self'
-    mail_combine: 'last'
-    memory_update: 'rnn'
-    mailbox_size: 1
-    combine_node_feature: True
-    dim_out: 100
-gnn:
-  - arch: 'identity'
-    time_transform: 'JODIE'
-train:
-  - epoch: 100
-    batch_size: 600
-    lr: 0.0001
-    dropout: 0.1
-    all_on_gpu: True
\ No newline at end of file

config/TGAT.yml

-sampling:
-  - layer: 2
-    neighbor: 
-      - 10
-      - 10
-    strategy: 'uniform'
-    prop_time: False
-    history: 1
-    duration: 0
-    num_thread: 32
-memory: 
-  - type: 'none'
-    dim_out: 0
-gnn:
-  - arch: 'transformer_attention'
-    layer: 2
-    att_head: 2
-    dim_time: 100
-    dim_out: 100
-train:
-  - epoch: 10
-    batch_size: 600
-    lr: 0.0001
-    dropout: 0.1
-    att_dropout: 0.1
-    all_on_gpu: True
\ No newline at end of file

config/TGN.yml

-sampling:
-  - layer: 1
-    neighbor: 
-      - 10
-    strategy: 'recent'
-    prop_time: False
-    history: 1
-    duration: 0
-    num_thread: 32
-memory: 
-  - type: 'node'
-    dim_time: 100
-    deliver_to: 'self'
-    mail_combine: 'last'
-    memory_update: 'gru'
-    mailbox_size: 1
-    combine_node_feature: True
-    dim_out: 100
-gnn:
-  - arch: 'transformer_attention'
-    layer: 1
-    att_head: 2
-    dim_time: 100
-    dim_out: 100
-train:
-  - epoch: 10
-    #batch_size: 100
-    # reorder: 16
-    lr: 0.0001
-    dropout: 0.2
-    att_dropout: 0.2
-    all_on_gpu: True
\ No newline at end of file

config/readme.yml

-sampling:
-  - layer: <number of layers to sample>
-    neighbor: <a list of integers indicating how many neighbors are sampled in each layer>
-    strategy: <'recent' that samples most recent neighbors or 'uniform' that uniformly samples neighbors form the past>
-    prop_time: <False or True that specifies wherether to use the timestamp of the root nodes when sampling for their multi-hop neighbors>
-    history: <number of snapshots to sample on>
-    duration: <length in time of each snapshot, 0 for infinite length (used in non-snapshot-based methods)
-    num_thread: <number of threads of the sampler>
-memory: 
-  - type: <'node', we only support node memory now>
-    dim_time: <an integer, the dimension of the time embedding>
-    deliver_to: <'self' that delivers the mails only to involved nodes or 'neighbors' that deliver the mails to neighbors>
-    mail_combine: <'last' that use the latest latest mail as the input to the memory updater>
-    memory_update: <'gru' or 'rnn'>
-    mailbox_size: <an integer, the size of the mailbox for each node>
-    combine_node_feature: <False or True that specifies whether to combine node features (with the updated memory) as the input to the GNN.
-    dim_out: <an integer, the dimension of the output node memory>
-gnn:
-  - arch: <'transformer_attention' or 'identity' (no GNN)>
-    layer: <an integer, number of layers>
-    att_head: <an integer, number of attention heads>
-    dim_time: <an integer, the dimension of the time embedding>
-    dim_out: <an integer, the dimension of the output dynamic node embedding>
-train:
-  - epoch: <an integer, number of epochs to train>
-    batch_size: <an integer, the batch size (of edges); for multi-gpu training, this is the local batchsize>
-    reorder: <(optional) an integer that is divisible by batch size the specifies how many chunks per batch used in the random chunk scheduling>
-    lr: <floating point, learning rate>
-    dropout: <floating point, dropout>
-    att_dropout: <floating point, dropout for attention>
-    all_on_gpu: <False or True that decides if the node/edge features and node memory are completely stored on GPU>
\ No newline at end of file

docs/Makefile

+# Minimal makefile for Sphinx documentation
+#
+
+# You can set these variables from the command line, and also
+# from the environment for the first two.
+SPHINXOPTS    ?=
+SPHINXBUILD   ?= sphinx-build
+SOURCEDIR     = source
+BUILDDIR      = build
+
+# Put it first so that "make" without argument is like "make help".
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+# Catch-all target: route all unknown targets to Sphinx using the new
+# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

docs/make.bat

+@ECHO OFF
+
+pushd %~dp0
+
+REM Command file for Sphinx documentation
+
+if "%SPHINXBUILD%" == "" (
+	set SPHINXBUILD=sphinx-build
+)
+set SOURCEDIR=source
+set BUILDDIR=build
+
+%SPHINXBUILD% >NUL 2>NUL
+if errorlevel 9009 (
+	echo.
+	echo.The 'sphinx-build' command was not found. Make sure you have Sphinx
+	echo.installed, then set the SPHINXBUILD environment variable to point
+	echo.to the full path of the 'sphinx-build' executable. Alternatively you
+	echo.may add the Sphinx directory to PATH.
+	echo.
+	echo.If you don't have Sphinx installed, grab it from
+	echo.https://www.sphinx-doc.org/
+	exit /b 1
+)
+
+if "%1" == "" goto help
+
+%SPHINXBUILD% -M %1 %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
+goto end
+
+:help
+%SPHINXBUILD% -M help %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
+
+:end
+popd

docs/source/conf.py

+# Configuration file for the Sphinx documentation builder.
+#
+# For the full list of built-in configuration values, see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+# -- Project information -----------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
+
+project = 'StarryGL'
+copyright = '2023, StarryGL Team'
+author = 'StarryGL Team'
+
+# -- General configuration ---------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
+
+extensions = []
+
+templates_path = ['_templates']
+exclude_patterns = []
+
+
+
+# -- Options for HTML output -------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
+
+html_theme = 'alabaster'
+html_static_path = ['_static']

docs/source/index.rst

+.. StarryGL documentation master file, created by
+   sphinx-quickstart on Sat Dec 23 01:02:22 2023.
+   You can adapt this file completely to your liking, but it should at least
+   contain the root `toctree` directive.
+
+Welcome to StarryGL's documentation!
+====================================
+
+.. toctree::
+   :maxdepth: 2
+   :caption: Contents:
+
+
+
+Indices and tables
+==================
+
+* :ref:`genindex`
+* :ref:`modindex`
+* :ref:`search`

starrygl/__init__.py

@@ -2,6 +2,8 @@ import torch
 
 import logging
 
+__version__ = "0.1.0"
+
 try:
     from .lib import libstarrygl_ops as ops
 except Exception as e:
@@ -12,4 +14,5 @@ try:
     from .lib import libstarrygl_ops_sampler as sampler_ops
 except Exception as e:
     logging.error(e)
-    logging.error("unable to import libstarrygl_sampler.so, some features may not be available.")
\ No newline at end of file
+    logging.error("unable to import libstarrygl_sampler.so, some features may not be available.")
+    
\ No newline at end of file