Skip to content

Commit

Permalink
skeleton of decision tree regressor added
Browse files Browse the repository at this point in the history
  • Loading branch information
gyrdym committed Nov 1, 2019
1 parent 4ed62d0 commit ff5e500
Show file tree
Hide file tree
Showing 6 changed files with 41 additions and 1 deletion.
7 changes: 6 additions & 1 deletion lib/src/di/dependencies.dart
Expand Up @@ -26,6 +26,8 @@ import 'package:ml_algo/src/math/randomizer/randomizer_factory.dart';
import 'package:ml_algo/src/math/randomizer/randomizer_factory_impl.dart';
import 'package:ml_algo/src/model_selection/data_splitter/data_splitter_factory.dart';
import 'package:ml_algo/src/model_selection/data_splitter/data_splitter_factory_impl.dart';
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor_factory.dart';
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor_factory_impl.dart';
import 'package:ml_algo/src/regressor/knn_regressor/knn_regressor_factory.dart';
import 'package:ml_algo/src/regressor/knn_regressor/knn_regressor_factory_impl.dart';

Expand Down Expand Up @@ -74,4 +76,7 @@ Injector get dependencies =>
(injector) => KnnRegressorFactoryImpl(
injector.getDependency<KernelFactory>(),
injector.getDependency<KnnSolverFactory>(),
));
))

..registerSingleton<DecisionTreeRegressorFactory>(
(_) => const DecisionTreeRegressorFactoryImpl());
@@ -0,0 +1,10 @@
import 'package:ml_algo/src/di/dependencies.dart';
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor.dart';
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor_factory.dart';

DecisionTreeRegressor createDecisionTreeRegressor() {
final regressorFactory = dependencies
.getDependency<DecisionTreeRegressorFactory>();

return regressorFactory.create();
}
@@ -0,0 +1,5 @@
import 'package:ml_algo/src/regressor/decision_tree_regressor/_helpers/create_decision_tree_regressor.dart';

abstract class DecisionTreeRegressor {
factory DecisionTreeRegressor() => createDecisionTreeRegressor();
}
@@ -0,0 +1,5 @@
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor.dart';

abstract class DecisionTreeRegressorFactory {
DecisionTreeRegressor create();
}
@@ -0,0 +1,10 @@
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor.dart';
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor_factory.dart';
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor_impl.dart';

class DecisionTreeRegressorFactoryImpl implements DecisionTreeRegressorFactory {
const DecisionTreeRegressorFactoryImpl();

@override
DecisionTreeRegressor create() => DecisionTreeRegressorImpl();
}
@@ -0,0 +1,5 @@
import 'package:ml_algo/src/regressor/decision_tree_regressor/decision_tree_regressor.dart';

class DecisionTreeRegressorImpl implements DecisionTreeRegressor {

}

0 comments on commit ff5e500

Please sign in to comment.