from athenet.sparsifying import sparsify_smallest_on_network
from athenet.algorithm import simple_neuron_deleter, simple_neuron_deleter2
from athenet.models import lenet
from athenet.data_loader import MNISTDataLoader



datasets = {
    "sender": [[(0.3, 0.75)],
               [(0.02, 1.0), (0.04, 1.0), (0.06, 1.0), (0.08, 1.0), (0.1, 1.0),
                (0.12, 1.0), (0.14, 1.0), (0.16, 1.0), (0.18, 1.0), (0.2, 1.0),
                (0.22, 1.0), (0.24, 1.0), (0.26, 1.0), (0.28, 1.0), (0.3, 1.0),
                (0.325, 1.0), (0.35, 1.0), (0.375, 1.0), (0.4, 1.0),
                (0.45, 1.0), (0.5, 1.0), (0.55, 1.0), (0.6, 1.0), (0.7, 1.0),
                (0.8, 1.0), (0.9, 1.0)]],
    "sender2": [[(0.3, 0.75)],
               [(0.02, 1.0), (0.04, 1.0), (0.06, 1.0), (0.08, 1.0), (0.1, 1.0),
                (0.12, 1.0), (0.14, 1.0), (0.16, 1.0), (0.18, 1.0), (0.2, 1.0),
                (0.22, 1.0), (0.24, 1.0), (0.26, 1.0), (0.28, 1.0), (0.3, 1.0),
                (0.325, 1.0), (0.35, 1.0), (0.375, 1.0), (0.4, 1.0),
                (0.45, 1.0), (0.5, 1.0), (0.55, 1.0), (0.6, 1.0), (0.7, 1.0),
                (0.8, 1.0), (0.9, 1.0)]],
    "rat":    [[50],
               [10, 20, 30, 40, 50, 60, 70, 80, 90],
               [5, 10, 15, 20, 25, 30, 35, 40, 45, 50]]
    }


algorithms = {
    "sender": simple_neuron_deleter,
    "sender2": simple_neuron_deleter2,
    "rat": sparsify_smallest_on_network
    }


def get_network(network_type):
    if network_type == "lenet":
        net = lenet()
        net.data_loader = MNISTDataLoader()
        return net
    raise NotImplementedError


def ok():
    print "[ \033[32mOK\033[39m ]"
@@ -213,16 +213,238 @@
 oader%0A%0A%0A
+%22%22%22%0A    Dataset - set (list) of configs given to algorithm as an input.%0A    %22datasets%22 is a dictionary from algorithm shortcut to list of available%0A    datasets.%0A    Do not forget to update help message after changing!%0A%22%22%22
 %0Adataset
@@ -1506,16 +1506,88 @@
     %7D%0A%0A%0A
+%22%22%22%0A    dictionary form algorithm shortcut to function to be called%0A%22%22%22%0A
 algorith
@@ -1748,16 +1748,145 @@
 _type):%0A
+    %22%22%22%0A        Returns a athenet.network of given type.%0A        :network_type: is a name of the type given as a string.%0A    %22%22%22%0A
     if n
@@ -2026,16 +2026,16 @@
 Error%0A%0A%0A
-
 def ok()
@@ -2036,16 +2036,82 @@
 f ok():%0A
+    %22%22%22%0A        prints a good looking %22OK%22 on the stdout.%0A    %22%22%22%0A
     prin

Add documentation of config