babao.models.tree package

Submodules

babao.models.tree.extremaModel module

The idea of that model is to find local extrema, then classify them as minimum/nop/maximum (-1/0/1) using a knn classifier (sklearn)

class babao.models.tree.extremaModel.ExtremaModel

Bases: babao.models.modelBase.ABCModel

A stupid simple model finding local extrema

dependencies_class = [<class 'babao.inputs.trades.krakenTradesInput.KrakenTradesXXBTZEURInput'>]

load()

Load the model from self.model_file

need_training = True

plot(since)

Plot the model predictions from ´since´ timestamp

predict(since)

Return a dataframe of prediction starting from ´since´ timestamp

save()

Save the model to self.model_file

train(since)

Train the model with data starting from ´since´ timestamp

Return the score of model.

babao.models.tree.macdModel module

Simple macd based model, with a very elegant algorithm (aka: brute-force)

class babao.models.tree.macdModel.MacdModel

Bases: babao.models.modelBase.ABCModel

Simple macd based model

dependencies_class = [<class 'babao.inputs.trades.krakenTradesInput.KrakenTradesXXBTZEURInput'>]

load()

Load the model from self.model_file

need_training = True

plot(since)

Plot the model predictions from ´since´ timestamp

predict(since)

Return a dataframe of prediction starting from ´since´ timestamp

save()

Save the model to self.model_file

train(since)

Train the model with data starting from ´since´ timestamp

Return the score of model.

babao.models.tree.tendencyModel module

Module contents