fix init

colinskow · Jun 6, 2017 · d83b06c · d83b06c
1 parent 7c3ebde
commit d83b06c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tflearn/__init__.py b/tflearn/__init__.py
@@ -37,7 +37,7 @@
 from .variables import variable, get_all_trainable_variable, \
     get_all_variables, get_layer_variables_by_name, get_layer_variables_by_scope
 from .objectives import categorical_crossentropy, binary_crossentropy, \
-    softmax_categorical_crossentropy, hinge_loss, mean_square
+    softmax_categorical_crossentropy, hinge_loss, mean_square, weighted_crossentropy
 from .metrics import Top_k, Accuracy, R2, top_k_op, accuracy_op, r2_op, Prediction_Counts
 
 # Direct layers inclusion