import sklearn_pmml.extensions.linear_regression

# TODO:
# import .extensions.svm
# import .extensions.random_forest

import sklearn_pmml.extensions.linear_regression
#import sklearn_pmml.extensions.logistic_regression
import sklearn_pmml.extensions.random_forest
# TODO:
# import .extensions.svm
# import .extensions.random_forest

Add randomforest to __init__.py