1
0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-28 13:48:56 +08:00

17 Commits

Author SHA1 Message Date
Ryan Schmukler
cf6192c81c fix(id3): fix panic on SplitAttribute being nil 2016-06-28 14:36:48 -04:00
Richard Townsend
7ba57fe6df trees: Handling FloatAttributes.
This patch adds:

	* Gini index and information gain ratio as
           DecisionTree split options;
	* handling for numeric Attributes (split point
           chosen naïvely on the basis of maximum entropy);
	* A couple of additional utility functions in base/
	* A new dataset (see sources.txt) for testing.

Performance on Iris performs markedly without discretisation.
2014-10-26 17:40:38 +00:00
Amit Kumar Gupta
1809a8b358 RandomForest returns error when fitting data with fewer features than the RandomForest plans to use
- BaseClassifier Predict and Fit methods return errors
- go fmt ./...

Conflicts:
	ensemble/randomforest.go
	ensemble/randomforest_test.go
	trees/tree_test.go
2014-08-22 13:39:29 +00:00
Amit Kumar Gupta
529b3bcaa5 Avoid renaming packages on import 2014-08-22 13:39:29 +00:00
Amit Kumar Gupta
947ee8380e Return error instead of panicking when unable to get confusion matrix 2014-08-22 13:39:29 +00:00
Amit Kumar Gupta
21bb2fc9fa Remove redundant import renames 2014-08-22 07:21:24 +00:00
Richard Townsend
47341b2869 base: Cleaned up duplicate Attribute resolution functions 2014-08-03 15:17:20 +01:00
Richard Townsend
c2d040af30 trees: merge from v2-instances 2014-08-03 15:17:13 +01:00
Niclas Jern
627a5537d3 Comments should be of the form "<Struct> ..." or "<MethodName> ..." 2014-07-18 13:48:28 +03:00
Niclas Jern
32f36f28c3 if block ends with a return statement -> drop this else and outdent its block 2014-07-18 13:20:46 +03:00
Remo Hertig
f77c1dcde0 use multiple return values instead of an array in InstancesTrainTestSplit 2014-06-06 21:33:17 +02:00
Richard Townsend
a6072ac9de Package documentation 2014-05-19 12:59:11 +01:00
Richard Townsend
45ca6063f1 Not sure if this bagging version is better or not
More more similar to "Attribute bagging:improving accuracy of classifier ensembles by using random feature subsets" (Brill)
2014-05-18 11:49:35 +01:00
Richard Townsend
26660e1470 Corrected a problem with pruning, actual ID3 decision tree type
Going to modify Bagging to select attributes on its own
2014-05-17 21:45:26 +01:00
Richard Townsend
12ace9def5 Identified source of the low accuracy 2014-05-17 20:37:19 +01:00
Richard Townsend
13c0dc3eba Reduced-error pruning 2014-05-17 18:06:01 +01:00
Richard Townsend
db3ac3c695 ID3 algorithm working 2014-05-17 17:28:51 +01:00